首页 > 代码库 > 【BZOJ】3319: 黑白树(并查集+特殊的技巧/-树链剖分+线段树)
【BZOJ】3319: 黑白树(并查集+特殊的技巧/-树链剖分+线段树)
http://www.lydsy.com/JudgeOnline/problem.php?id=3319
以为是模板题就复习了下hld。。。。。。。。。。。。。。。。。。。。。。。。。。。。。
然后nlg^2n被tle成翔了。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。
然后看题解QAQ,,,这。。。
神题做法。。。待会再写。。。
tle的hld:
#include <cstdio>#include <cstring>#include <cmath>#include <string>#include <iostream>#include <algorithm>#include <queue>#include <set>#include <map>using namespace std;typedef long long ll;#define rep(i, n) for(int i=0; i<(n); ++i)#define for1(i,a,n) for(int i=(a);i<=(n);++i)#define for2(i,a,n) for(int i=(a);i<(n);++i)#define for3(i,a,n) for(int i=(a);i>=(n);--i)#define for4(i,a,n) for(int i=(a);i>(n);--i)#define CC(i,a) memset(i,a,sizeof(i))#define read(a) a=getint()#define print(a) printf("%d", a)#define dbg(x) cout << (#x) << " = " << (x) << endl#define error(x) (!(x)?puts("error"):0)#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }const int N=1e6+10;int ihead[N], cnt;struct dat { int next, to, id; }e[N<<1];void add(int u, int v, int id) { e[++cnt].next=ihead[u]; ihead[u]=cnt; e[cnt].to=v; e[cnt].id=id; e[++cnt].next=ihead[v]; ihead[v]=cnt; e[cnt].to=u; e[cnt].id=id;}int n, son[N], top[N], fa[N], sz[N], root, id[N], tot, ed[N], dep[N];void dfs1(int x, int f) { sz[x]=1; int y; fa[x]=f; dep[x]=dep[f]+1; rdm(x, i) if(f!=(y=e[i].to)) { dfs1(y, x); sz[x]+=sz[y]; if(sz[y]>sz[son[x]]) son[x]=y; }}void dfs2(int x, int tp) { top[x]=tp; id[x]=++tot; if(son[x]) dfs2(son[x], tp); rdm(x, i) if(e[i].to!=fa[x] && e[i].to!=son[x]) dfs2(e[i].to, e[i].to);}void geted(int x, int t) { ed[id[x]]=t; rdm(x, i) if(e[i].to!=fa[x]) geted(e[i].to, e[i].id);}#define lc x<<1#define rc x<<1|1#define lson l, mid, lc#define rson mid+1, r, rc#define MID (l+r)>>1struct T { int mx; bool tag; }t[N<<2];void upd(int x, int r) { t[x].tag=1; t[x].mx=max(t[x].mx, r);}void pushdown(int x, int mid, int r) { if(t[x].tag) t[x].tag=0, upd(lc, mid), upd(rc, r);}void pushup(int x) { t[x].mx=max(t[lc].mx, t[rc].mx); }void update(int L, int R, int l=1, int r=tot, int x=1) { if(L<=l && r<=R) { upd(x, r); return; } int mid=MID; pushdown(x, mid, r); if(L<=mid) update(L, R, lson); if(mid<R) update(L, R, rson); pushup(x);}int ask(int L, int R, int l=1, int r=tot, int x=1) { if(L<=l && r<=R) return t[x].mx; int mid=MID, ret=0; pushdown(x, mid, r); if(L<=mid) ret=max(ret, ask(L, R, lson)); if(mid<R) ret=max(ret, ask(L, R, rson)); return ret;}void ask(int x) { int fx=top[x], pos=0; while(fx!=root) { pos=ask(id[fx], id[x]); if(pos!=0) break; x=fa[fx]; fx=top[x]; } if(pos==0 && x!=root) pos=ask(id[top[x]], id[x]); printf("%d\n", ed[pos]);}void change(int x, int y) { int fx=top[x], fy=top[y]; while(fx!=fy) { if(dep[fx]<dep[fy]) { swap(fx, fy); swap(x, y); } update(id[fx], id[x]); x=fa[fx]; fx=top[x]; } if(x==y) return; if(dep[x]<dep[y]) swap(x, y); update(id[y]+1, id[x]);}int main() { read(n); int m=getint(); for1(i, 1, n-1) { int u=getint(), v=getint(); add(u, v, i); } root=1; dfs1(root, 0); dfs2(root, root); geted(root, 0); while(m--) { int cc=getint(); if(cc==1) ask(getint()); else { int u=getint(), v=getint(); change(u, v); } } return 0;}
Description
给定一棵树,边的颜色为黑或白,初始时全部为白色。维护两个操作:
1.查询u到根路径上的第一条黑色边的标号。
2.将u到v 路径上的所有边的颜色设为黑色。
Notice:这棵树的根节点为1
Input
第一行两个数n,m分别表示点数和操作数。
接下来n-? 1行,每行2个数u,v.表示一条u到v的边。
接下来m行,每行为以下格式:
1 v 表示第一个操作
2 v u 表示第二种操作
Output
对于每个询问,输出相应答案。如果不存在,输出0。
Sample Input
5 4
1 2
1 3
2 4
2 5
1 2
2 2 3
1 3
1 4
1 2
1 3
2 4
2 5
1 2
2 2 3
1 3
1 4
Sample Output
0
2
1
2
1
HINT
对于 100% 的数据:n,m<=10^6
Source
【BZOJ】3319: 黑白树(并查集+特殊的技巧/-树链剖分+线段树)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。