Directing Edges(拓扑排序)

题目链接: Directing Edges

大致题意:

给定n个顶点, m条有向边的图(不一定连通), 这m条边有些边是已经确定好方向的, 有一些是还没有确定的, 你需要确定所有未确定方向的边(让他们从a->b or b->a), 使得图中不出现环.

若可以实现, 输出YES和一种可行解, 否则输出NO

解题思路:

拓扑排序的应用, 先不管所有未确定方向的边, 只计算所有确定方向的边, 若此时图中无环, 则我们一定也可以存在某种方式使得最终的图中也无环.

分析: 如果原先图中无环, 则一定存在一个合法的拓扑序, 此时若a和b之间存在一条未确定的边, 只需要按照拓扑序从小指向大即可. 若a的拓扑序小于b的拓扑序, 则当a的入度为0时, 会将a->b这条边的约束除去, 不会影响到最终图的环路情况.

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= n; ++i)
using namespace std;
typedef long long ll;
const int N = 2E5 + 10;
int n, m;
int du[N], level[N]; //level表示拓扑排序大小
vector<int> edge[N];
bool topo() {
    queue<int> q; int cou = 0;
    rep(i, n) if (!du[i]) q.push(i);
    while (!q.empty()) {
        auto op = q.front(); q.pop(); level[op] = ++cou;
        for (auto& to : edge[op]) {
            if (!--du[to]) q.push(to);
        }
    }
    return cou == n;
}
int main()
{
    int t; cin >> t;
    while (t--) {
        scanf("%d %d", &n, &m);
        rep(i, n) edge[i].clear(), du[i] = 0;
        vector<pair<int, int>> v[2]; //分别存确定和未确定的边
        rep(i, m) {
            int k, a, b; scanf("%d %d %d", &k, &a, &b);
            v[k].push_back({ a, b });
            if (k) edge[a].push_back(b), du[b]++;
        }
        if (topo()) {
            printf("YES\n");
            for (auto& op : v[1]) printf("%d %d\n", op.first, op.second);
            for (auto& op : v[0]) {
                int a = op.first, b = op.second;
                if (level[a] > level[b]) swap(a, b); //按照拓扑序 小->大
                printf("%d %d\n", a, b);
            }
        }
        else printf("NO\n");
    }
    return 0;
}

END