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;
}