Codeforces Round #589 (Div. 2) D. Complete Tripartite（STL+思维）

题目链接


思路：其实没必要去建图，转换一下思维，如果是在同一个集合他们有什么特征？存边以后会发现相同集合的他们的连接点是一样的，比如样例里的2连接着{1、4、5、6}，3连接着{1、4、5、6}那么2和3肯定是同一集合，只要用set存一下即可，有一个坑点就是如果一个点没有连边的时候要特判。

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+1;
set<int>s[maxn];
map<set<int>,int>p;
int n,m,cnt=0;
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;++i)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		s[u].insert(v);s[v].insert(u); 
	}
	for(int i=1;i<=n;++i)
	{
		if(s[i].size()==0) {printf("-1\n");return 0;}
		if(!p[s[i]]) p[s[i]]=++cnt;
	}
	if(cnt!=3) {printf("-1\n");return 0;}
	for(int i=1;i<=n;++i)
	printf("%d ",p[s[i]]);
}