【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 500500
using 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;
}