【AC梦工厂】广工华工联合训练1 A - Comunicating the Tibet

传送门：https://vjudge.net/contest/188332#problem/A

题目大意：

给定一个（n,m）图，需要在每个点上给一个不超过k的值，使得相邻的点的值不相同，不存在时输出-1

题解

一开始想法是先求联通分支，再topo给点赋值，但看到别人的代码后发现更简单的思路是，直接在dfs的过程中给点赋值，因为在dfs的过程中已经相当于是topo的过程了。

代码

#include <iostream>
#include <bits/stdc++.h>

using namespace std;
int s[400005];
vector<int>g[400005];
int n,m,k;
void dfs(int x,int cnt)
{
    s[x]=cnt;
    for(int i=0;i<g[x].size();i++)
    {
        if(s[g[x][i]]==-1)
            dfs(g[x][i],cnt%k+1);
    }
}
int main()
{
    fill(s,s+40005,-1);
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<m;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
    if(m!=0&&k==1){
        printf("-1\n");
        return 0;
    }
    for(int i=1;i<=n;i++)
    {
        if(s[i]==-1) dfs(i,1);
        printf("%d\n",s[i]);
    }
    return 0;
}