POJ 3687 逆向拓扑排序

题意:n个球,m个关系, a b 代表 a 比 b 轻, 按编号1~n输出每个球是第几大,要使得编号小的球排名尽量小. 例如:5->6->1 4->3->2 1号球可以是第3,4,5,6重 这里要选第三重, 那么二号球就只能是最重的一个了,即6. 输出答案为:3 6 5 4 1 2

思路:反向建立拓扑排序, 则以重的球入度为0开始, 用优先队列每次弹出入度为零编号最大的球即可

#define INF 0x3f3f3f3f
#define maxn 204
#define Pair pair<int, int>
#define mem(a, b) memset(a, b, sizeof(a))
using namespace std;
int T;
int n, m;
vector<int> g[maxn];
int ans[maxn];
int in[maxn];
int he[maxn];
int topsort()
{
    int sum = 0, temp = n;
    priority_queue<int> que;

    for (int i = 1; i <= n; ++i) {
        if (in[i] == 0) que.push(i);
    }

    while ( !que.empty() ) {
        int x = que.top();
        sum++;
        ans[x] = temp--;
        que.pop();
        for (int j = 0; j < g[x].size(); ++j) {
            int v = g[x][j];
            if (--in[v] == 0) {
                que.push(v);
            }
        }
    }
    if (sum < n) return 0;
    return 1;
}
int main()
{
    cin >> T;
    while (T--) {
        for(int i = 1; i <= n; ++i) {
            g[i].clear();
            in[i] = 0;
        }
        mem(ans, 0);
        cin >> n >>m;
        for (int i = 1; i <= m; ++i) {
            int a, b;
            cin >> a >> b;
            g[b].push_back(a);
            in[a]++;
        }
        int flag = topsort();
        if(flag)
            for (int i = 1; i <= n; ++i) {
                if(i-1) cout << " ";
                cout << ans[i];
            }
        else {
            cout <<"-1";
        }
        cout << endl;
    }
    return 0;
}