【bzoj4551】树 并查集

4551: [Tjoi2016&Heoi2016]树

TimeLimit: 20 Sec  Memory Limit: 128 MB
Submit: 380  Solved: 234
[Submit][Status][Discuss]

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

 

小（lao）结（sao）：调了半小时发现自己数据出错了qwq

写程序时不要随便改变自己的码风，强改一不小心就会出错。

 

题解：

并查集

对于每一次将点标记，我们可以理解为形成了一个新的联通块，这个联通块里的点的询问回答就是这个点！！！(真的很巧妙！)但是在线的话我们发现处理不了，所以我们用离线的做法，从后往前将连通块连起来。

 

上代码:

#include<iostream>#include<cstdio>#include<cstring>#define N 500500using namespace std;int v[N],next[N],head[N],fa[N],ba[N],cnt[N],q[N],d[N],ans[N];int n,m,tot=0,o=0;char c[N];void add_edge(int x,int y){v[++tot]=y;next[tot]=head[x];head[x]=tot;}int getfa(int x){return x == fa[x]?x:fa[x]=getfa(fa[x]);}void dfs(int x,int yeye)//防止回路{    for (int i=head[x];i;i=next[i])    {        int y=v[i];        if (y != yeye)        {            ba[y]=x;            if (!cnt[y]) fa[y]=x;            dfs(y,x);        }    }}int main(){    memset(cnt,0,sizeof(cnt));    scanf("%d%d",&n,&m);    for (int i=0;i<n-1;i++)    {        int x,y;        scanf("%d%d",&x,&y);        add_edge(x,y);        add_edge(y,x);    }    for (int i=0;i<m;i++)    {        scanf(" %c%d",&c[i],&d[i]);        if (c[i] == 'C') cnt[d[i]]++;//记录下点被重复标记的次数    }    for (int i=1;i<=n;i++) fa[i]=i,ba[i]=i;    dfs(1,0);    for (int i=m-1;i>=0;i--)    {        if (c[i] == 'Q')            ans[o++]=getfa(d[i]);        else        {            cnt[d[i]]--;            if (!cnt[d[i]]) fa[d[i]]=ba[d[i]];        }    }    for (int i=o-1;i>=0;i--) printf("%d\n",ans[i]);    return 0;}