蓝桥杯ADV-169 士兵排队问题 【拓扑排序】

题目描述

　　有Ｎ个士兵(1≤Ｎ≤26)，编号依次为Ａ,Ｂ,Ｃ,…，队列训练时，指挥官要把一些士兵从高到矮一次排成一行，但现在指挥官不能直接获得每个人的身高信息，只能获得“P1比P2高”这样的比较结果(P1、P2∈Ａ,Ｂ,Ｃ,…,Ｚ,记为 P1>P2)，如”Ａ>Ｂ”表示Ａ比Ｂ高。
　　请编一程序，根据所得到的比较结果求出一种符合条件的排队方案。
　　（注：比较结果中没有涉及的士兵不参加排队）

输入要求

　　比较结果从文本文件中读入（文件由键盘输入），每个比较结果在文本文件中占一行。

输出要求

　　若输入数据无解，打印“No Answer!”信息，否则从高到矮一次输出每一个士兵的编号，中间无分割符，并把结果写入文本文件中，文件由键盘输入：

样例输入

A>B
B>D
F>D

样例输出

AFBD

解题思路 

拓扑排序。《数据结构（严蔚敏）》中对拓扑排序的解释：

A>B，看做一条从A -> B的边，

（1）选择一个没有入边的点（兵），说明没有比他高的，输出这个点（兵）

（2）删除这个点以及它的一切出边

重复（1）（2），直至全部顶点均已输出，或者当前图中不存在无前驱的顶点（这些点（兵）的身高无法比较）。

AC代码

#include <bits/stdc++.h>
using namespace std;
int main()
{
	char a[3];
	bool book[26];
	memset(book,false,sizeof book);
	set<int > s_out[26];	
	set<int > s_in[26];
	while(scanf("%s",a)!=EOF) {
		s_out[a[0]-'A'].insert(a[2]-'A');
		s_in[a[2]-'A'].insert(a[0]-'A');
		book[a[0]-'A'] = true;
		book[a[2]-'A'] = true;
	}
	
	queue<int > q;
	for(int i=0;i<26;i++)
		if(s_in[i].size() == 0 && book[i])
			q.push(i);	//记录所有入度为0的点
	
	char ans[26];
	int k=0;
	while(!q.empty()) {
		int j = q.front();
		for(set<int>::iterator it=s_out[j].begin();it!=s_out[j].end();it++) {
			int to = *it;
			s_in[to].erase(j);
			if(s_in[to].size() == 0)
				q.push(to);
		}
		s_out[j].clear();
		ans[k++] = (char)j+'A';
		q.pop();
	} 
	
	int i;
	for(i=0;i<26;i++)
		if(s_in[i].size() > 0)
			break;		
	if(i == 26)
		for(int j=0;j<k;j++)
			printf("%c",ans[j]);
	else
		printf("No Answer!");
}