其他分享
首页 > 其他分享> > H106OJ | 士兵排队问题

H106OJ | 士兵排队问题

作者:互联网

士兵排队问题

Description

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

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

(注:比较结果中没有涉及的士兵不参加排队)

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