BZOJ 4551: [Tjoi2016&Heoi2016]树

4551: [Tjoi2016&Heoi2016]树

Time Limit: 20 Sec  Memory Limit: 128 MB

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序，用线段树维护。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<vector>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int,int> pr;
const double pi=acos(-1);
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define per(i,n,a) for(int i=n;i>=a;i--)
#define Rep(i,u) for(int i=head[u];i;i=Next[i])
#define clr(a) memset(a,0,sizeof(a))
#define pb push_back
#define mp make_pair
#define fi first
#define sc second
#define pq priority_queue
#define pqb priority_queue <int, vector<int>, less<int> >
#define pqs priority_queue <int, vector<int>, greater<int> >
#define vec vector
ld eps=1e-9;
ll pp=1000000007;
ll mo(ll a,ll pp){if(a>=0 && a<pp)return a;a%=pp;if(a<0)a+=pp;return a;}
ll powmod(ll a,ll b,ll pp){ll ans=1;for(;b;b>>=1,a=mo(a*a,pp))if(b&1)ans=mo(ans*a,pp);return ans;}
void fre() { freopen("c://test//input.in", "r", stdin); freopen("c://test//output.out", "w", stdout); }
//void add(int x,int y,int z){ v[++e]=y; next[e]=head[x]; head[x]=e; cost[e]=z; }
int dx[5]={0,-1,1,0,0},dy[5]={0,0,0,-1,1};
ll read(){ ll ans=0; char last=' ',ch=getchar();
while(ch<'0' || ch>'9')last=ch,ch=getchar();
while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
if(last=='-')ans=-ans; return ans;
}
const int N=100005,M=N+N;
int fa[N],tf[N],v[M],Next[M],head[N],ans[N],f[N],e,ansnu,n,m;
struct node{
    int f,u;
}Q[N];
void add(int x,int y){ v[++e]=y; Next[e]=head[x]; head[x]=e;}
int getfa(int x){
    if (fa[x]==x) return x;
    else return fa[x]=getfa(fa[x]);
}
void dfs(int u,int f_){
    for (int i=head[u];i;i=Next[i]){
        if (v[i]==f_) continue;
        dfs(v[i],u);
    }
    if (!f[u]) fa[u]=f_; tf[u]=f_;
}
int main(){
    n=read(),m=read(); int x,y; char ch[5];
    for (int i=1;i<n;i++) {
        x=read(),y=read();
        add(x,y); add(y,x);
    }
    for (int i=1;i<=n;i++) fa[i]=i; f[1]=1;
    for (int i=1;i<=m;i++) {
        scanf("%s",ch);
        if (ch[0]=='C'){
            Q[i].f=0; Q[i].u=read();
            f[Q[i].u]++;
        } else Q[i].f=1,Q[i].u=read();
    }
    dfs(1,0);
    for (int i=m;i>=1;i--){
        if (Q[i].f) ans[++ansnu]=getfa(Q[i].u);
        else { 
            f[Q[i].u]--;
            if (!f[Q[i].u]) fa[Q[i].u]=tf[Q[i].u];
        }
    }
    for (int i=ansnu;i>=1;i--){
        printf("%d
",ans[i]);
    }
    return 0;
}