b_wy_奥运会真实排名(拓扑排序)

T组数据,每组数据表示有 m 个人给出自己看到的运动员的冲线排名,
问能不能通过每组数据中每个观众给出的排名算出运动员的真实排名,
能就输出排名,否则输出 NO

思路:...

#include<bits/stdc++.h>
using namespace std;
void f() {
    int n, m;
    cin >> n >> m;
    vector<int> in(n+1, 0);
    vector<int> g[n+1];
    for (int i = 0; i < m; i++) {
        int c, pre = -1;
        cin >> c;
        while (c--) {
            int u; cin >> u;
            if (pre != -1) {
                g[pre].push_back(u);
                in[u]++;
            }
            pre = u;
        }
    }
    queue<int> q;
    for (int i = 1; i <= n; i++) {
        if (in[i] == 0) {
            q.push(i);
        }
    }
    vector<int> ans;
    while (!q.empty()) {
        if (q.size() != 1) {
            break;
        }
        int u = q.front();
        ans.push_back(u), q.pop();
        for (int v : g[u]) {
            if (--in[v] == 0) {
                q.push(v);
            }
        }
    }
    if (ans.size() == n) {
        for (int i = 0; i < ans.size(); i++) {
            cout << ans[i] << " \n"[i == n-1];
        }
    } else {
        cout << "NO\n";
    }
}
int main() {
    std::ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
    int t; cin >> t;
    while (t--) {
        f();
    }
    return 0;
}
上一篇:git stash


下一篇:BZOJ 3714 Kuglarz