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