H106OJ | 士兵排队问题

士兵排队问题

Description

有Ｎ个士兵(1≤Ｎ≤26)，编号依次为Ａ,Ｂ,Ｃ,…，队列训练时，指挥官要把一些士兵从高到矮一次排成一行，但现在指挥官不能直接获得每个人的身高信息，只能获得“P1比P2高”这样的比较结果(P1、P2∈Ａ,Ｂ,Ｃ,…,Ｚ,记为 P1>P2)，如”Ａ>Ｂ”表示Ａ比Ｂ高。

请编一程序，根据所得到的比较结果求出一种符合条件的排队方案。

（注：比较结果中没有涉及的士兵不参加排队）

Input

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

Output

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

Sample Input 1

A>B
B>D
F>B
F<A

Sample Output 1

AFBD

Hint

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

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 30, M = 1010;

int e[M], ne[M], h[N], idx;
char s[N];
int only[N][N], vis[N], ans[N], in[N];
int sum, cnt;
queue<int> q;
priority_queue<int, vector<int>, greater<int> > pq; 

void add(int a, int b) {
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx++;
}

int main() {
	memset(h, -1, sizeof h);
	while(~scanf("%s", s)) {
		int a = s[0]-64, b = s[2]-64;
		if(!only[a][b]) {
			add(a, b);
			in[b] ++; 
			only[a][b] = 1;
			if(!vis[a]) cnt++, vis[a] = 1; 
			if(!vis[b]) cnt++, vis[b] = 1;
		}
	}
	for(int i = 1;i <= 26;i ++) 
		if(vis[i] && !in[i]) q.push(i);
	while(!q.empty()) { 
		int t = q.front();
		q.pop();
		ans[++sum] = t;
		
		for(int i = h[t];i != -1;i = ne[i]) {
			int j = e[i];
			--in[j];
			if(!in[j]) pq.push(j);
		}
		
		while(!pq.empty()) {
			int t = pq.top();
			pq.pop();
			q.push(t);
		}
	}
	if(cnt == sum) 
		for(int i = 1; i <= sum; i++) 
			cout << (char)(ans[i]+64);
	else puts("No Answer!");
}

解题思路：

运用了拓扑排序，入度为0的点作为拓扑排序的起点，需要注意没涉及的士兵不参加排队，所以建立一个标记数组记录参加排队的士兵，剩下的就是裸的拓扑排序。

标签：idx,int,排队,++,vis,H106OJ,士兵
来源： https://blog.csdn.net/weixin_54786344/article/details/123620625