【BZOJ】3319: 黑白树（并查集+特殊的技巧/-树链剖分+线段树）

http://www.lydsy.com/JudgeOnline/problem.php?id=3319

以为是模板题就复习了下hld。。。。。。。。。。。。。。。。。。。。。。。。。。。。。

然后nlg^2n被tle成翔了。。。。。。。。。。。。。。。。。。。。。。。。。。。。。。

然后看题解QAQ，，，这。。。

神题做法。。。待会再写。。。（upd：【BZOJ】3319: 黑白树）

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