codeforces 1217D - Coloring Edges（topu）

链接

题意：

有n个点m条边组成的有向图，要给边涂色，使得不存在一个由相同颜色组成的环。

思路：

如果图中本来就没有环，全部涂成一种颜色即可。
有环的情况时，由于一个组成一个环的边必定有两种边，编号小的–>编号大的、编号大的–>编号小的。把这两种边涂成两种颜色即可。

参考代码：

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

const int N=5e3+5;

int head[N],cnt;
struct node{
    int u,v,id,nxt;
}edge[N<<1];

void add_edge(int u,int v,int id){
    edge[++cnt].u=u;
    edge[cnt].v=v;
    edge[cnt].nxt=head[u];
    edge[cnt].id=id;
    head[u]=cnt;
}

int ind[N];
queue<int>q;
int ans[N];
bool vis[N];
int sz;
void topu(){
    while (!q.empty()){
        int u=q.front();
        q.pop();
        sz++;
        vis[u]= true;
        for(int i=head[u];~i;i=edge[i].nxt){
            int v=edge[i].v;
            ans[edge[i].id]=1;
            ind[v]--;
            if(!ind[v]){
                q.push(v);
            }
        }
    }
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    memset(head,-1, sizeof(head));
    for(int i=1,u,v;i<=m;i++){
        scanf("%d%d",&u,&v);
        ind[v]++;
        add_edge(u,v,i);
    }
    for(int i=1;i<=n;i++){
        if(!ind[i]){
            q.push(i);
        }
    }
    sz=0;
    topu();
    printf("%d\n",sz==n?1:2);
    for(int i=1;i<=m;i++){
        if(sz==n)ans[edge[i].id]=1;
        else ans[edge[i].id]=edge[i].u<edge[i].v?1:2;
    }
    for(int i=1;i<=m;i++){
        printf("%d%c",ans[i],i==m?'\n':' ');
    }
    return 0;
}