试题 有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
样例输出 AFBD
思路:裸的拓扑排序,简单描述一下: 记录每个节点的入度,每次从栈中挑选一个入度为0的点并把其下级节点的入度-1,酱如果下级节点入度将为0,随即将其入栈(有些点可能会一直被压在栈底,没关系,因为没有先后关系的节点顺序是无所谓的),重复此操作,直至栈空. 如果此时还存在入度大于0的节点,则无解(这个可以通过记录出栈的节点个数判断).
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<cstring>
#include<string>
#include<vector>
#include<cmath>
#include<map>
#define mem(a,b) memset(a,b,sizeof(a))
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
const double esp = 1e-7;
const int ff = 0x3f3f3f3f;
map<int,int>::iterator it;
vector<int> mp[52];
stack<int> q;
int ans[52],num[52],vis[52];
int main()
{
char in[5];
while(scanf(" %s",in) !=EOF)
{
int i = in[0]-'A',j = in[2]-'A';
vis[i] = vis[j] = 1;
mp[i].push_back(j);
num[j]++;
}
int cnt = 0;
for(int i = 0;i< 26;i++)
{
if(!num[i]&&vis[i])
q.push(i);
}
while(!q.empty())
{
int tmp = q.top();
q.pop();
ans[++cnt] = tmp;
int k = mp[tmp].size();
for(int i = 0;i< k;i++)
{
int id = mp[tmp][i];
if(!(--num[id]))
q.push(id);
}
}
for(int i = 0;i< 26;i++)
if(num[i]> 0)
{
cout<<"No Answer!"<<endl;
return 0;
}
for(int i = 1;i<= cnt;i++)
printf("%c",ans[i]+'A');
return 0;
}