蓝桥杯 士兵排队问题(拓扑排序)

试题   有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;
}

    原文作者:拓扑排序
    原文地址: https://blog.csdn.net/nka_kun/article/details/79505778
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞