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

倒序并查集1A
这应该是经典问题了

#include<bits/stdc++.h> using namespace std;#define For(i,n) for(int i=1;i<=n;i++)#define Fork(i,k,n) for(int i=k;i<=n;i++)#define Rep(i,n) for(int i=0;i<n;i++)#define ForD(i,n) for(int i=n;i;i--)#define RepD(i,n) for(int i=n;i>=0;i--)#define Forp(x) for(int p=pre[x];p;p=next[p])#define Forpiter(x) for(int &p=iter[x];p;p=next[p])  #define Lson (o<<1)#define Rson ((o<<1)+1)#define MEM(a) memset(a,0,sizeof(a));#define MEMI(a) memset(a,0x3f,sizeof(a));#define MEMi(a) memset(a,128,sizeof(a));#define MEMx(a,b) memset(a,b,sizeof(a));#define INF (0x3f3f3f3f)#define F (1000000007)#define pb push_back#define mp make_pair#define fi first#define se second#define vi vector<int> #define pi pair<int,int>#define SI(a) ((a).size())#define Pr(kcase,ans) printf("Case #%d: %lld\n",kcase,ans);#define PRi(a,n) For(i,n-1) cout<<a[i]<<' '; cout<<a[n]<<endl;#define PRi2D(a,n,m) For(i,n) { \                        For(j,m-1) cout<<a[i][j]<<' ';\                        cout<<a[i][m]<<endl; \                        } #pragma comment(linker, "/STACK:102400000,102400000")#define ALL(x) (x).begin(),(x).end()typedef long long ll;typedef long double ld;typedef unsigned long long ull;ll mul(ll a,ll b){return (a*b)%F;}ll add(ll a,ll b){return (a+b)%F;}ll sub(ll a,ll b){return ((a-b)%F+F)%F;}void upd(ll &a,ll b){a=(a%F+b%F)%F;}inline int read(){    int x=0,f=1; char ch=getchar();    while(!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}    while(isdigit(ch)) { x=x*10+ch-'0'; ch=getchar();}    return x*f;}#define MAXN (100000+10)class bingchaji{public:    int father[MAXN],n,cnt;    void mem(int _n)    {        n=cnt=_n;        For(i,n) father[i]=i;    }    int getfather(int x)     {        if (father[x]==x) return x;        return father[x]=getfather(father[x]);    }    void unite(int x,int y)    {        x=getfather(x);        y=getfather(y);        if (x^y) {            --cnt;            father[x]=y;        }    }    bool same(int x,int y)    {        return getfather(x)==getfather(y);    }}S;vi edges[MAXN];int fa[MAXN];void dfs(int x,int f){    fa[x]=f;    Rep(i,SI(edges[x])) {        if (edges[x][i]^f) dfs(edges[x][i],x);    }}int b[MAXN],a[MAXN],ans[MAXN];int is_tak[MAXN]={0};int main(){//  freopen("bzoj4551.in","r",stdin);//  freopen(".out","w",stdout);    int n=read(),q=read();    For(i,n-1) {        int u=read(),v=read();        edges[u].pb(v);edges[v].pb(u);    }    dfs(1,0);    For(i,q) {        char s[10];        scanf("%s %d",s,&a[i]);        b[i]=(s[0]=='Q');        if (!b[i]) is_tak[a[i]]++;    }    S.mem(n);    is_tak[1]++;    For(i,n) if (!is_tak[i]) S.unite(i,fa[i]);    ForD(i,q) {        if (b[i]) ans[i]=S.getfather(a[i]);        else if (--is_tak[a[i]]==0){            S.unite(a[i],fa[a[i]]);        }    }    For(i,q) if (b[i]) printf("%d\n",ans[i]);    return 0;}