codeforces Codeforces Round #656 (Div. 3) E. Directing Edges (拓扑排序 )

链接: E. Directing Edges
题意:
给一个图,有的边方向没有确定,是否可以给这些边加上方向是其变成一个无环图。
思路:
先对有向的边进行拓扑排序,判断是否有环,如果有环就是 NO, 否则就让 拓扑序小的点指向拓扑序大的点(有向无环图一定是拓扑序小的指向大的)。
代码:

#include 
using namespace std;
typedef long long ll;
const int maxn = 2e5+7;
const int mod=1e9+7;
int head[maxn],num,deg[maxn],n,m,fl,T,pos[maxn];
vector< int > ans;
vector< pair<int ,int > > vec;
struct node{
       int to,next;
}e[maxn];
void add(int u,int v){
     e[num].to=v;
     e[num].next=head[u];
     head[u]=num++;
}
void topo(){
     queue<int >que;
     for(int i=0;i<n;i++){
         if(deg[i]==0){
            que.push(i);
         }
     }
     while(que.size()){
          int x=que.front();que.pop();
          ans.push_back(x);
          pos[x]=ans.size();
          for(int i=head[x];i!=-1;i=e[i].next){
              deg[e[i].to]--;
              if(deg[e[i].to]==0){
                 que.push(e[i].to);
              }
          }
     }
}
int main (){
    cin>>T;
    while(T--){
         cin>>n>>m;
         int k=0;num=0;
         for(int i=0;i<=n;i++) deg[i]=0,head[i]=-1;
         ans.clear();
         vec.clear();
         for(int i=0,x,u,v;i<m;i++){
            scanf("%d%d%d",&x,&u,&v);
            u--,v--;
            vec.push_back(make_pair(u,v));
            if(x==1){
              add(u,v);
              deg[v]++;
            }
        }
        topo();
        if(ans.size()!=n){
          printf ("NO\n");
          continue;
        }
        printf ("YES\n");
        for(auto p:vec){
            int u=p.first,v=p.second;
            if(pos[u]>pos[v]) swap(u,v);
            printf ("%d %d\n",u+1,v+1);
        }
    }
}

 

你可能感兴趣的:(codeforces)