蓝桥 士兵排队问题 拓扑排序

算法提高 士兵排队问题  

时间限制：1.0s   内存限制：256.0MB

    

试题

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

输入要求

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

输出要求

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

样例输入

A>B
B>D
F>D

样例输出

AFBD

#include <bits/stdc++.h>

using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=26;
typedef long long ll;

int in[maxn];//入度
vector<int> next[maxn];//指向的 
vector<int> ans;
int vis[maxn];
char str[10];
stack<int> s;

void TopSort(){
	int cnt=0;//排序结束，如果cnt小于节点数，那么存在环 
	while(!s.empty()){
		int k=s.top();
		s.pop();
		ans.push_back(k);
		cnt++;
		for(int i=0;i<next[k].size();i++){
			int t=next[k][i];
			in[t]--; 
			if(in[t]==0)	s.push(t);
		}	
	}
	for(int i=0;i<maxn;i++)
		if(vis[i])	cnt--;
	if(cnt==0){
		for(int i=0;i<ans.size();i++)
			printf("%c",ans[i]+'A');
		printf("\n");
	}
	else	printf("No Answer!\n");
}

int main(){
	memset(in,0,sizeof(in));
	memset(vis,0,sizeof(vis));
	ans.clear(); 
    while(scanf("%s",str) !=EOF ){
    	int u=str[0]-'A',v=str[2]-'A';
    	vis[u]=vis[v]=1;
    	if(str[1]=='>'){
    		next[u].push_back(v);
    		in[v]++;
    	}
    	else{
    		next[v].push_back(u);
    		in[u]++;
    	}
    }
    for(int i=0;i<maxn;i++)
    	if(vis[i]&&in[i]==0)//入度为零的入栈 
    		s.push(i);
   	TopSort();
	return 0;
}