D Coloring Edges (拓扑排序/dfs 判环)

题目链接

题目大意:

对一个有向图的边进行染色,要求一个环内的边颜色不能相同。

思路:

如果这个图无环,那么全部输出1,如果存在环,那么就让大节点指向小节点的边染成2,否则染成1。最多只有两种染色情况,因为一个环和另一个环之间不会存在联系,所以只要保证每一个环是合法的就行了。
那么现在就只剩下判环了,可以拓扑排序或dfs。dfs时的判环和无向图不同,这一点需要注意。
DFS:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int N=5e3+10;
struct node
{
    
    
    int to,next;
}e[N];
int flag=0,head[N],cnt=0,a[N],b[N],n,m,vis[N];
void addedge(int u,int v)
{
    
    
    e[++cnt].to=v;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void dfs(int now)
{
    
    
    vis[now]=-1;//表示正在访问
    for(int i=head[now];i;i=e[i].next)
    {
    
    
        int to=e[i].to;
        if(vis[to]==0) dfs(to);
        else if(vis[to]==-1)
        {
    
    
            flag=1;
            return;
        }
    }
    vis[now]=1;//该节点子节点全部访问完。
}
int main()
{
    
    
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
    
    
        scanf("%d%d",&a[i],&b[i]);
        addedge(a[i],b[i]);
    }
    for(int i=1;i<=n;i++)
    {
    
    
        if(vis[i]==0) dfs(i);
    }
    if(!flag)
    {
    
    
        printf("1\n");
        for(int i=1;i<=m;i++) printf("1 ");
        printf("\n");
    }
    else
    {
    
    
        printf("2\n");
        for(int i=1;i<=m;i++)
        {
    
    
            if(a[i]>b[i]) printf("2 ");
            else printf("1 ");
        }
        printf("\n");
    }
    return 0;
}

topsort:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int N=5e3+10;
struct node
{
    
    
    int to,next;
}e[N];
int num=0,head[N],cnt=0,deg[N],t[N],a[N],b[N],n,m;
void addedge(int u,int v)
{
    
    
    e[++cnt].to=v;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void topsort()
{
    
    
    queue<int>q;
    for(int i=1;i<=n;i++) if(deg[i]==0) q.push(i);
    while(!q.empty())
    {
    
    
        int now=q.front();q.pop();
        t[now]=++num;
        for(int i=head[now];i;i=e[i].next)
        {
    
    
            int to=e[i].to;
            deg[to]--;
            if(deg[to]==0) q.push(to);
        }
    }
}
int main()
{
    
    
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
    
    
        scanf("%d%d",&a[i],&b[i]);
        deg[b[i]]++;
        addedge(a[i],b[i]);
    }
    topsort();
    if(num==n)
    {
    
    
        printf("1\n");
        for(int i=1;i<=m;i++) printf("1 ");
        printf("\n");
    }
    else
    {
    
    
        printf("2\n");
        for(int i=1;i<=m;i++)
        {
    
    
            if(a[i]>b[i]) printf("2 ");
            else printf("1 ");
        }
        printf("\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/amazingee/article/details/107565486