题目链接
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;
}