[蓝桥杯][ADV-169][算法提高][vip][拓扑排序]士兵排队问题

蓝桥杯 ADV-169 算法提高 士兵排队问题

题目描述
有Ｎ个士兵(1≤Ｎ≤26)，编号依次为 Ａ,Ｂ,Ｃ,…，队列训练时，指挥官要把一些士兵从高到矮一次排成一行，但现在指挥官不能直接获得每个人的身高信息，只能获得“P1比P2高”这样的比较 结果(P1、P2∈Ａ,Ｂ,Ｃ,…,Ｚ,记为 P1> P2)，如”Ａ> Ｂ”表示Ａ比Ｂ高。
请编一程序，根据所得到的比较结果求出一种符合条件的排队方案。
（注：比较结果中没有涉及的士兵不参加排队）
输入
比较结果从文本文件中读入（文件由键盘输入），每个比较结果在文本文件中占一行。
输出
若输入数据无解，打印“No Answer!”信息，否则从高到矮一次输出每一个士兵的编号，中间无分割符，并把结果写入文本文件中，文件由键盘输入
样例输入
A> B
B> D
F> D
样例输出
AFBD

---

这是一道裸的拓扑排序。

拓扑排序的解释：在一个有向图中，对所有的节点进行排序，要求没有一个节点指向它前面的节点。
先统计所有节点的入度，对于入度为0的节点就可以分离出来，然后把这个节点指向的节点的入度减一。
一直做改操作，直到所有的节点都被分离出来。
如果最后不存在入度为0的节点，那就说明有环，不存在拓扑排序，也就是很多题目的无解的情况。

//下面是我常用的拓扑排序的框架
void top_ord() {
    int cnt = 0;
    queue<int >q;
    vector<int >ans;
    for (int i = 0; i < 200; i++) {
        if (in[i] == 0) {
            q.push(i);
        }
        if (in[i] != -1) {
            cnt++;
        }
    }
    while (!q.empty()) {
        int p = q.front();
        q.pop();
        ans.push_back(p);
        for (int i = 0; i < edge[p].size(); i++) {
            int tem = edge[p][i];
            in[tem] --;
            if (in[tem] == 0) {
                q.push(tem);
            }
        }
    }
    if (ans.size() == cnt) {
        for (int i = 0; i < ans.size(); i++) {
            cout << char(ans[i]);
        }
    } else {
        cout << "No  Answer!";
    }
    return;
}

---

解题代码如下：

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;

vector<int >edge[200];
int in[200];
void top_ord() {
    int cnt = 0;
    queue<int >q;
    vector<int >ans;
    for (int i = 0; i < 200; i++) {
        if (in[i] == 0) {
            q.push(i);
        }
        if (in[i] != -1) {
            cnt++;
        }
    }
    while (!q.empty()) {
        int p = q.front();
        q.pop();
        ans.push_back(p);
        for (int i = 0; i < edge[p].size(); i++) {
            int tem = edge[p][i];
            in[tem] --;
            if (in[tem] == 0) {
                q.push(tem);
            }
        }
    }
    if (ans.size() == cnt) {
        for (int i = 0; i < ans.size(); i++) {
            cout << char(ans[i]);
        }
    } else {
        cout << "No  Answer!";
    }
    return;
}
int main () {
    char a, b;
    fill(in, in + 200, -1);
    while(~scanf("%c> %c ",&a, &b)) {
        edge[a].push_back(b);
        if (in[a] == -1) {
            in[a] = 0;
        }
        if (in[b] == -1) {
            in[b] = 0;
        }
        in[b]++;
    }
    top_ord();
    return 0;
}