BZOJ4551: [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

 

丝帛数据结构题，考虑使用dfs序来解决这个问题，那么第一个操作对应区间内答案与val取max，第二个操作对应单点询问，用棵线段树打标记就行了。

#include<cstdio>
#include<cctype>
#include<queue>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
const int BufferSize=1<<16;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
	if(head==tail) {
		int l=fread(buffer,1,BufferSize,stdin);
		tail=(head=buffer)+l;
	}
	return *head++;
}
inline int read() {
    int x=0,f=1;char c=Getchar();
    for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=Getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=100010;
int n,m,first[maxn],next[maxn<<1],to[maxn<<1],e;
void AddEdge(int u,int v) {
	to[++e]=v;next[e]=first[u];first[u]=e;
	to[++e]=u;next[e]=first[v];first[v]=e;
}
int dep[maxn],st[maxn],en[maxn],cnt;
int better(int x,int y) {return dep[x]<dep[y]?y:x;}
void dfs(int x,int fa) {
	dep[x]=dep[fa]+1;st[x]=++cnt;
	ren if(to[i]!=fa) dfs(to[i],x);
	en[x]=cnt;
}
int setv[maxn<<2];
void update(int o,int l,int r,int ql,int qr,int val) {
	if(ql<=l&&r<=qr) setv[o]=better(setv[o],val);
	else {
		int mid=l+r>>1,lc=o<<1,rc=lc|1;
		if(ql<=mid) update(lc,l,mid,ql,qr,val);
		if(qr>mid) update(rc,mid+1,r,ql,qr,val);
	}
}
int query(int o,int l,int r,int p) {
	if(l==r) return setv[o];
	int mid=l+r>>1,lc=o<<1,rc=lc|1,ans=setv[o];
	if(p<=mid) ans=better(ans,query(lc,l,mid,p));
	else ans=better(ans,query(rc,mid+1,r,p));
	return ans;
}
int main() {
	n=read();m=read();
	rep(i,2,n) AddEdge(read(),read());
	dfs(1,0);update(1,1,n,1,n,1);
	while(m--) {
		char c=Getchar();
		while(!isalpha(c)) c=Getchar();
		int x=read();
		if(c=='Q') printf("%d\n",query(1,1,n,st[x]));
		else update(1,1,n,st[x],en[x],x);
	}
	return 0;
}