CF-1385E——Directing Edges

题目链接
https://codeforces.com/problemset/problem/1385/E
思路
      题目大意就是给你一个图,有无向边也有有向边,那么能不能把所有无向边都变成某个方向的有向边形成有向无环图,如果不行输出“NO”,可以输出"YES"并输出所有有向边(包括已有的有向边和把无向边变成有向边的有向边)。
做法:
      首先对所有有向边进行拓扑排序并存放这个排序的序列(桶排思想存放),如果已经成环,直接输出“NO”即可,反之输出“YES”,然后对于每个无向边都将起点终点通过拓扑序列进行比较,前面的点是不依赖后面的点的,所以在拓扑序列中即可确定无向边的指向,需要声明的是,本题输出答案一定是不唯一的,题意也有说明。
      存图方式除了利用基本的前向星存储所有有向边之外,还需要另开两个pair<int,int>类型的vector v1,v2分别存放无向边和有向边便于后面的检测和输出。
c++代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5+5;
vector<int> v[N];
vector<pair<int,int>> v1,v2;
int n,m,e[N],cnt;
int s[N];
int vis[N];
void topsort(){
    
    
	queue<int> q;
	for(int i=1;i<=n;i++){
    
    
		if(vis[i]==0) q.push(i);
	}
	while(!q.empty()){
    
    
		int u=q.front();
		q.pop();
		cnt++;
		s[u]=cnt;
		for(int i=0;i<v[u].size();i++){
    
    
			vis[v[u][i]]--;
			if(vis[v[u][i]]==0) q.push(v[u][i]);
		}
	}
}
int main()
{
    
    
    int t;
    scanf("%d",&t);
    while(t--)
    {
    
    
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;i++)
        {
    
    
           vis[i]=0;
           s[i]=0;
           v[i].clear();
        }
        v1.clear();
        v2.clear();
        for(int i=1;i<=m;i++)
        {
    
    
            int x,b;
            scanf("%d%d%d",&x,&e[i],&b);
            if(x)
            {
    
    
                vis[b]++;
                v[e[i]].push_back(b);
                v2.push_back(make_pair(e[i],b));
            }
            else v1.push_back(make_pair(e[i],b));
        }
        cnt=0;
        topsort();
        if(cnt<n)
        {
    
    
            printf("NO\n");
        }
        else
        {
    
    
            printf("YES\n");
            for(int i=0;i<v2.size();i++)
            {
    
    
                printf("%d %d\n",v2[i].first,v2[i].second);
            }
            for(int i=0;i<v1.size();i++)
            {
    
    
                if(s[v1[i].first]>s[v1[i].second]) swap(v1[i].first,v1[i].second);
                printf("%d %d\n",v1[i].first,v1[i].second);
            }
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Star_Platinum14/article/details/112161700