题目大意:
对一个有向图的边进行染色,要求一个环内的边颜色不能相同。
思路:
如果这个图无环,那么全部输出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;
}