codeforces C. News Distribution

题目链接：https://codeforces.com/contest/1167/problem/C
题目大意：有n个人，m个组。每个组中所有人都是朋友。**需要注意的是，一个人可以属于多个组。**然后，从一个人开始传递消息，他会传递给他所有的朋友，他的朋友会传递给他所有的朋友，直到朋友中没有人不知道这个消息。求这个人的消息能被多少人知道（包括自己）。求1 ~ n每个人都传递一次，每个人对应的值。
题目思路：其实很简单，就是自己bfs一次，然后看自己生成树中有多少个节点就行。首先 题目的意思就是一种广播的形式，利用bfs肯定可以得到正确结果。现在，我们需要做的就是改进。因为m组给出的信息，每组给出的都是一棵树。我们需要把这些树合并，然后得到一个n个人的关系图（图是无向的），然后bfs就行，（对于每次bfs完之后，我们可以标记已经遍历过的节点，这些节点的结果都是一样的。）bfs完之后对树中所有的节点进行赋值，值都是树的节点个数。下次遇到树中的节点就可以bfs了。对于如何生成一个完整图，我们可以将每组的边都插入到图中，而每组给出的数据应该是一个完全图，但是由于是无向图，所以我们只需要一棵树就行。因此，我们可以将第一点作为根节点，其余点都指向根。然后将每条边都加入到图中。（因为，一个点可能出现在多个组中，因此很有可能会是多重边，因此我们需要删除这些边，故利用set进行去重）
总结一下就是：
1）处理数据，将数据转化成图，用邻接链表存储，set进行去重处理。
2）执行下面的步骤：对于i，1 <= i <= n ,如果mark[i] = 0,bfs(i)。输出mark[i]。否则直接输出mark[i]。
对于bfs。普通的bfs就行，只是最后我们需要对刚刚bfs遍历过的节点进行赋值，大小就是刚刚遍历过的节点个数。

#include <bits/stdc++.h>
using namespace std;

const int maxn = 500005;
int mark[maxn];
set<int> G[maxn];
void bfs(int u){
	vector<int> v;
	int cnt = 0;
	
	queue<int> q;
	q.push(u);
	mark[u] = 1;
	
	while(!q.empty()){
		int a = q.front();q.pop();
		v.push_back(a);cnt++;
		set<int>::iterator iter = G[a].begin();
		while(iter != G[a].end()){
			int b = *iter;
			if(!mark[b])q.push(b), mark[b] = 1;
			iter++;
		}
	}
	for(int i=0;i<cnt;i++){
		mark[v[i]] = cnt;
	}
}
int main(){
    ios::sync_with_stdio(false);
	int n, m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)mark[i] = 0;
	for(int i=0;i<m;i++){
		int k;cin>>k;
		if(k == 0)continue;
		if(k == 1){cin>>k;continue;}
		int a, b; cin>>a;
		for(int j=1;j<k;j++){
			cin>>b;
			G[a].insert(b);
			G[b].insert(a);
		}
	}
	for(int i=1;i<=n;i++){
		if(!mark[i])bfs(i);
		cout<<mark[i]<<" ";
	}
}

附：这个程序处理的不太好，其实这个可以用并查集来处理。大家可以去codeforces上看并查集的处理方式。