蓝桥杯 算法提高 士兵排队问题 拓扑排序 java

试题 算法提高 士兵排队问题

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

分析：这个题目输入的格式是文本的输入的话自己可以看一下输入格式
而这个题目主要的话考察拓扑排序，直接写代码

import java.util.*;

public class ADV_169士兵排队问题 {
    static int car[][] = new int[26][26];
    static int vis[] = new int[26];
    static Stack<Integer> stack = new Stack<Integer>();
    static HashSet<Character> set = new HashSet<Character>();
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int x,y;
		while(sc.hasNext()) {
			String str = sc.nextLine();
			char arr[] = str.toCharArray();
			set.add(arr[0]);
			set.add(arr[2]);
			x=arr[0]-'A';
			y=arr[2]-'A';
		    car[x][y]=1;
		}
		for (int i = 0; i <26; i++) {
			if(vis[i]==1)continue;
			boolean  bool = dfs(i);
			if(!bool) {
				System.out.println("No Answer!");
				return;
			}
		}
		while(!stack.empty()) {
			char c = (char) (stack.pop()+'A');
			int n = set.size();
			set.add(c);
			if(n == set.size()) {
				System.out.print(c);
			}
			
		}

	}
	private static boolean dfs(int i) {
		vis[i] = -1;
		for (int j = 0; j <26; j++) {
			if(car[i][j]>0) {
				if(vis[j]<0)return false;
				if(vis[j] == 0 &&  dfs(j) == false)return false;
			}
		}
		stack.add(i);
		vis[i] = 1;
		return true;
	}

}