【题解】牛客OI周赛1-提高组 A.分组 dfs

链接：https://www.nowcoder.com/acm/contest/199/A
来源：牛客网

---

将认识关系转化为图中的边。dfs这张图，对每一个没有被访问过的点，将它标记为源点的反色，回溯的时候统计每个点有多少同色相邻点，个数等于2时将其颜色转换。

#include
#include
const int N=1e5+10,M=2e5+10;
int n,m,tot,vis[N],hd[N];
struct Edge{
	int v,nx;
}e[M<<1];
void dfs(int u)
{
	int cnt=0;
	for(int i=hd[u];~i;i=e[i].nx)
	    if(!vis[e[i].v])vis[e[i].v]=vis[u]^3,dfs(e[i].v);
	for(int i=hd[u];~i;i=e[i].nx)
	    cnt+=(vis[e[i].v]==vis[u]);
	if(cnt==2)vis[u]^=3;
}
void add(int u,int v)
{
	e[tot].v=v;
	e[tot].nx=hd[u];
	hd[u]=tot++;
}
int main()
{
	//freopen("in.txt","r",stdin);
	memset(hd,-1,sizeof(hd));
	scanf("%d%d",&n,&m);
    int u,v;
	for(int i=1;i<=m;i++)
    	scanf("%d%d",&u,&v),add(u,v),add(v,u);
    for(int i=1;i<=n;i++)
    	if(!vis[i])vis[i]=1,dfs(i);
    for(int i=1;i<=n;i++)
        printf("%d ",vis[i]);
    puts("");
    return 0;
}

总结

初看比赛结果发现这题只有1人满分，然后就被吓着了。在想会不会是并查集，或者在图中找环，后来发现这里的认识关系不会传递，且节点度数不超过3，肯定会存在一种解。