CodeForces 1167 C News Distribution 并查集

TP

题意: 有n个人和m个群, 群内的人都是好友, 好友的好友也是好友, 问如果告诉x一个消息, 他把这个消息发给所有的好友, 问这个消息会被多少人知道.

思路: 并查集, 如果两个人在同一个群中就让他们所在的集合合并, sz[x]维护以x为祖先节点的集合大小.

路径压缩一定写递归的! 迭代多难写!!

代码:

#include<bits/stdc++.h>
#define fuck(x) std::cout<<"["<<#x<<"->"<<x<<"]"<<endl;
using namespace std;
typedef long long ll;

const int M=5e5+5;
const int inf=1e9+5;
const int mod=1e9+7;

int fa[M];
int sz[M];

int findfa(int x) {
    return fa[x] == x ? x : fa[x] = findfa(fa[x]);
}


int main() {
    ios::sync_with_stdio(0);
    int n,m;
    cin>>n>>m;
    for(int i=1; i<=n; i++) {
        fa[i]=i;
        sz[i]=1;
    }
    for(int i=0; i<m; i++) {
        int num;
        cin>>num;
        int x,y;
        for(int j=0; j<num; j++) {
            if(j==0)
                cin>>x;
            else {
                cin>>y;
                int fx=findfa(x);
                int fy=findfa(y);
                if(fx!=fy) {
                    fa[fy]=fx;
                    sz[fx]+=sz[fy];
                }
            }
        }
    }
    for(int i=1; i<=n; i++)
        cout<<sz[findfa(i)]<<" ";
    puts("");
    return 0;
}