首页 > 代码库 > 【bzoj4551】[Tjoi2016&Heoi2016]树 离线处理+并查集
【bzoj4551】[Tjoi2016&Heoi2016]树 离线处理+并查集
题目描述
在2016年，佳媛姐姐刚刚学习了树，非常开心。现在他想解决这样一个问题：给定一颗有根树（根为1），有以下两种操作：1. 标记操作：对某个结点打上标记（在最开始，只有结点1有标记，其他结点均无标记，而且对于某个结点，可以打多次标记。）2. 询问操作：询问某个结点最近的一个打了标记的祖先（这个结点本身也算自己的祖先）你能帮帮他吗?
输入
输入第一行两个正整数N和Q分别表示节点个数和操作次数接下来N-1行，每行两个正整数u,v(1≤u,v≤n)表示u到v有一条有向边接下来Q行，形如“oper num”oper为“C”时表示这是一个标记操作,oper为“Q”时表示这是一个询问操作对于每次询问操作，1 ≤ N, Q ≤ 100000。
输出
输出一个正整数，表示结果
样例输入
5 5
1 2
1 3
2 4
2 5
Q 2
C 2
Q 2
Q 5
Q 3
样例输出
1
2
2
1
题解
在线的话可以树剖，然而我选择了离线处理+并查集。
加标记比较难搞，我们可以换一种思路，先把所有标记加进来，再从后往前删掉。
每个节点的f为最近的有标记的祖先，这可以在dfs中直接实现。
然后从后往前处理，如果是修改则删标记，删为0时直接将该点的f赋为f[fa]。
如果是查询，直接记录find即可。
理论时间复杂度O(αN)，然而这么慢也是醉了。
另外听说暴力可过。
#include <cstdio>
#include <algorithm>
#define N 100010
using namespace std;
int head[N] , to[N << 1] , next[N << 1] , cnt , fa[N] , t[N] , f[N] , opt[N] , p[N] , ans[N];
char str[5];
void add(int x , int y)
{
to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt;
}
void dfs(int x , int last)
{
if(t[x]) last = x;
f[x] = last;
int i;
for(i = head[x] ; i ; i = next[i])
if(to[i] != fa[x])
fa[to[i]] = x , dfs(to[i] , last);
}
int find(int x)
{
return x == f[x] ? x : f[x] = find(f[x]);
}
int main()
{
int n , m , i , x , y;
scanf("%d%d" , &n , &m);
for(i = 1 ; i < n ; i ++ ) scanf("%d%d" , &x , &y) , add(x , y) , add(y , x);
t[1] = 1;
for(i = 1 ; i <= m ; i ++ )
{
scanf("%s%d" , str , &p[i]);
if(str[0] == ‘C‘) opt[i] = 1 , t[p[i]] ++ ;
}
dfs(1 , 0);
for(i = m ; i >= 1 ; i -- )
{
if(opt[i])
{
t[p[i]] -- ;
if(!t[p[i]]) f[p[i]] = f[fa[p[i]]];
}
else ans[i] = find(p[i]);
}
for(i = 1 ; i <= m ; i ++ ) if(!opt[i]) printf("%d\n" , ans[i]);
return 0;
}
【bzoj4551】[Tjoi2016&Heoi2016]树 离线处理+并查集
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。