【Codeforces 1217D】Coloring Edges（构造）

题目链接

无环时答案为$1$。比赛时我大胆猜了一下有环时答案必为$2$，凭感觉瞎搞了一个算法：
1.拓扑排序，去掉的边标蓝色。如果没有入度不为$0$的点，结束。
2.取一个入度不为$0$的点，将它的入边标红色然后删去，回到步骤1。

当时抱着试试的心态交了一发，结果A掉了…因为$n$很小所以我没多想，每次暴力找入度不为$0$的点，写成了$O(n^2+m)$。如果用链表维护一下入度不为$0$的点就是$O(n+m)$。

 
在Announcement的评论区里看到了几个非常巧妙的解法，都是写起来很方便而且$O(n+m)$的：
1.如果$u>v$就把$(u,v)$标红，否则标蓝。因为从起点走过一个环回到起点的过程中，经过的点序号不可能始终单增或者单降，所以环边不会全部同色。
2.DFS时新边标红，返祖边标蓝。一个环必然要由新边和返祖边组合而成，故环边不会全部同色。
 

const int N=100005;
struct ex{int num,next,col;}map[N];
int n,m,head[N],d[N],len=1;
void link(int x,int y)
{
  len++;map[len].num=y;map[len].next=head[x];head[x]=len;
  len++;map[len].num=x;map[len].next=head[y];head[y]=len;
  d[y]++;
}
queue<int>q;
void TOPO(int X)
{
  int i,j,x,y;
  if(X==-1)
    {
      for(i=1;i<=n;i++)
		if(!d[i]) q.push(i);
    }
  else q.push(X);
  while(!q.empty( ))
    {
      x=q.front( );q.pop( );
      for(i=head[x];i;i=map[i].next)
		{
		  if(i&1) continue;
		  if(map[i].col) continue;
		  y=map[i].num;
		  d[y]--;
		  if(!d[y]) q.push(y);
		  map[i].col=1;
		}
    }
}
int check( )
{
  for(int i=1;i<=n;i++)
    if(d[i]) return i;
  return 0;
}
int main( )
{
  int i,j,x,y;
  n=read( );m=read( );
  for(i=1;i<=m;i++)
    {
      x=read( );y=read( );
      link(x,y);
    }
  TOPO(-1);
  while((x=check( ))!=0)
    {
      for(i=head[x];i;i=map[i].next)
		{
		  if(!(i&1)) continue;
		  if(map[i^1].col) continue;
		  map[i^1].col=2;d[x]--;
		}
      TOPO(x);
    }
  int ans=0;
  for(i=2;i<=len;i+=2)
    ans=max(ans,map[i].col);
  printf("%d\n",ans);
  for(i=2;i<=len;i+=2)
    printf("%d ",map[i].col);
  return 0;
}