算法提高 士兵排队问题
时间限制:1.0s 内存限制:256.0MB
试题
有N个士兵(1≤N≤26),编号依次为A,B,C,…,队列训练时,指挥官要把一些士兵从高到矮一次排成一行,但现在指挥官不能直接获得每个人的身高信息,只能获得“P1比P2高”这样的比较结果(P1、P2∈A,B,C,…,Z,记为 P1>P2),如”A>B”表示A比B高。
请编一程序,根据所得到的比较结果求出一种符合条件的排队方案。
(注:比较结果中没有涉及的士兵不参加排队)
请编一程序,根据所得到的比较结果求出一种符合条件的排队方案。
(注:比较结果中没有涉及的士兵不参加排队)
输入要求
比较结果从文本文件中读入(文件由键盘输入),每个比较结果在文本文件中占一行。
输出要求
若输入数据无解,打印“No Answer!”信息,否则从高到矮一次输出每一个士兵的编号,中间无分割符,并把结果写入文本文件中,文件由键盘输入:
样例输入
A>B
B>D
F>D
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;
}