Java实现 蓝桥杯VIP 算法提高 士兵排队问题

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

import java.util.LinkedList;
import java.util.Scanner;


public class 士兵排列问题 {//这道题的输入是文件输入，
	                     //与手动输入有些不一样
	 					//自己可以稍微改一下看看测试用例
	static boolean[] visit = new boolean[26];
	static boolean[][] dep = new boolean[26][26];
	static int[] in = new int[26];
	static char[] ans = new char[26];
	
	public static void topSort(){//拓扑排序
		LinkedList<Integer> q = new LinkedList<Integer>();
		int n = 0;
		for(int i=0;i<26;++i){
			if(visit[i] == true){
				n++;
				if(in[i] == 0)
					q.add(i);
			}
		}
		
		int index = 0;
		while(!q.isEmpty()){
			//队列首元素出队列
			int f = q.pollFirst();
			n--;
			ans[index++] = (char) (f + 'A');
			for(int i=0;i<26;++i){
				if(visit[i] && dep[f][i]){
					in[i]--;
					if(in[i] == 0)
						q.add(i);
				}
			}
		}
		if(n != 0){
			System.out.println("No Answer!");
		}else{
			for(int i=0;i<index;++i){
				System.out.print(ans[i]);
			}
		}
	}
	
    public static void main(String[] args) {
    	Scanner sc = new Scanner(System.in);
    	char a,op,b;
    	int x,y;
    	//int count = 0;
        while(sc.hasNext()){
        	
        	String str = sc.nextLine();
        	//count++;
         	char[] arr = str.toCharArray();
         	a=arr[0];
         	op=arr[1];
         	b=arr[2];
         	x = a-'A';
         	y = b-'A';
         	visit[x] = visit[y] = true;
         	if(dep[x][y] == false){
         		dep[x][y] = true;
         		in[y]++;
         	}
         	
        }
    	topSort();
    }

}