Codeforces Round #541 (Div. 2) F Asya And Kittens

题意

给你n-1对,相邻的在一块,选了这两个数,这两个集合就可以看成一个集合了,问符合条件的序列。

很容易想到并查集的合并操作,对于一次合并相当于把两个集合变成了一块

#include<ext/rope>库中 有个rope 算是一个可持久化数组,支持插入删除操作,并且支持+=-运算符

内部采用分块链表实现,复杂度基本sqrt(n)

#include<bits/stdc++.h>
#include<ext/rope>
using namespace __gnu_cxx;
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int maxn=1e6+50;
const ll inf=0x3f3f3f3f3f3f3f3fLL;
rope<int>a[maxn];
int fa[maxn];
int n;
int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int main()
{
	scanf("%d",&n);
    for(int i=1;i<=n;i++){
        a[i].push_back(i);
        fa[i]=i;
    }
    for(int i=1;i<n;i++){
        int x,y; scanf("%d%d",&x,&y);
        x=find(x),y=find(y);
        a[x]+=a[y];
        fa[y]=x;
    }
    for(int i=1;i<=n;i++){
        if(find(i)==i){
            for(int j=0;j<a[i].size();j++)
            	printf("%d ", a[i][j]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/wzazzy/article/details/87920998