【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)>>1
struct 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

Sample Output

0
2
1

HINT

对于    100%    的数据：n，m<=10^6

Source