BZOJ 4551 [Tjoi2016&Heoi2016]树 树链剖分

Description

在2016年，佳媛姐姐刚刚学习了树，非常开心。现在他想解决这样一个问题：给定一颗有根树（根为1），有以下

两种操作：1. 标记操作：对某个结点打上标记（在最开始，只有结点1有标记，其他结点均无标记，而且对于某个

结点，可以打多次标记。）2. 询问操作：询问某个结点最近的一个打了标记的祖先（这个结点本身也算自己的祖

先）你能帮帮他吗?

Input

输入第一行两个正整数N和Q分别表示节点个数和操作次数接下来N-1行，每行两个正整数u,v(1≤u,v≤n)表示u到v

有一条有向边接下来Q行，形如“opernum”oper为“C”时表示这是一个标记操作,oper为“Q”时表示这是一个询

问操作对于每次询问操作，1 ≤ N, Q ≤ 100000。

Output

输出一个正整数，表示结果

Sample Input

5 5
1 2
1 3
2 4
2 5
Q 2
C 2
Q 2
Q 5
Q 3

Sample Output

1
2
2
1

HINT

 新加数据9组(By HFLSyzx ),未重测--2016.8.2

Source

传送门

水了一道树链剖分……

太空虚了= =

子树更新，每次保留深度更大的一个结点。。

其实完全都不用树链剖分了，dfs序就好了的说……

打标记的时候忘记判断了深度更大的节点结果WA了一次。

#include<bits/stdc++.h>
using namespace std;
int read(){
    int x=0;char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x;
}
const int 
	N=100005;
int n,Q,Ecnt,cnt;
struct Tree{
	int pre,son,dep,top,sz,tid,MAX;
}tree[N];
struct Segment{
	int mark;
}seg[N<<2];
struct Edge{
	int next,to;
}E[N<<1];int head[N];
void add(int u,int v){
	E[++Ecnt].next=head[u];
	E[Ecnt].to=v;
	head[u]=Ecnt;
}
void dfs1(int u,int pre,int deep){
	tree[u].pre=pre;
	tree[u].dep=deep;
	tree[u].son=0;
	tree[u].sz=1;
	int maxs=0;
	for (int i=head[u];i;i=E[i].next){
		int v=E[i].to;
		if (v==pre) continue;
		dfs1(v,u,deep+1);
		tree[u].sz+=tree[v].sz;
		if (maxs<tree[v].sz) maxs=tree[v].sz,tree[u].son=v;
	}
}
void dfs2(int u,int ancestor){
	tree[u].tid=++cnt;
	tree[u].top=ancestor;
	tree[u].MAX=cnt;
	if (tree[u].son) dfs2(tree[u].son,ancestor);
	tree[u].MAX=max(tree[u].MAX,tree[tree[u].son].MAX);
	for (int i=head[u];i;i=E[i].next){
		int v=E[i].to;
		if (v==tree[u].pre || v==tree[u].son) continue;
		dfs2(v,v);
		tree[u].MAX=max(tree[u].MAX,tree[v].MAX);
	}
}
void down(int u){
	if (seg[u].mark){
		int l=u<<1,r=u<<1|1;
		if (tree[seg[u].mark].dep>tree[seg[l].mark].dep)
			seg[l].mark=seg[u].mark;
		if (tree[seg[u].mark].dep>tree[seg[r].mark].dep)
			seg[r].mark=seg[u].mark;
		seg[u].mark=0;
	}
}
void update(int id,int l,int r,int gl,int gr,int num){
	if (l>=gl && r<=gr){
		if (tree[seg[id].mark].dep<tree[num].dep)
			seg[id].mark=num;
		return;
	}
	down(id);
	int mid=(l+r)>>1;
	if (gl<=mid) update(id<<1,l,mid,gl,gr,num);
	if (gr>mid) update(id<<1|1,mid+1,r,gl,gr,num);
}
int query(int id,int l,int r,int g){
	if (l==r) return seg[id].mark;
	down(id);
	int mid=(l+r)>>1;
	if (g<=mid) return query(id<<1,l,mid,g);
		else return query(id<<1|1,mid+1,r,g);
}
int main(){
	n=read(),Q=read();
	int x,y;Ecnt=0;
	for (int i=1;i<n;i++){
		x=read(),y=read();
		add(x,y),add(y,x);
	}
	tree[0].dep=-1;
	dfs1(1,-1,0);
	cnt=0,dfs2(1,1);
	update(1,1,n,1,n,1);
	char opt[2];int num;
	while (Q--){
		scanf("%s",opt);
		num=read();
		if (opt[0]=='C') update(1,1,n,tree[num].tid,tree[num].MAX,num);
			else printf("%d\n",query(1,1,n,tree[num].tid));
	}
	return 0;
}