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
1 2
1 3
2 4
2 5
Q 2
C 2
Q 2
Q 5
Q 3
Sample Output
1
2
2
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;
}