2103-士兵排列问题 ZCMU

题目描述:

有Ｎ个士兵(1≤Ｎ≤26)，编号依次为 Ａ,Ｂ,Ｃ,…，队列训练时，指挥官要把一些士兵从高到矮一次排成一行，但现在指挥官不能直接获得每个人的身高信息，只能获得“P1比P2高”这样的比较 结果(P1、P2∈Ａ,Ｂ,Ｃ,…,Ｚ,记为 P1>P2)，如”Ａ>Ｂ”表示Ａ比Ｂ高。
请编一程序，根据所得到的比较结果求出一种字典序最小的排队方案。
（注：比较结果中没有涉及的士兵不参加排队）

输入:

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

输出:

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

样例:

A>B

B>D

F>D

样例结果:

ABFD

思路:拓扑排序+优先队列

代码:

#include<bits/stdc++.h>
using namespace std;
#define MAX 30
int mark[MAX],in[MAX],ans[MAX];
int main()
{
    int i,c,n,it1,it2;
    char a,x,b;
    memset(mark,0,sizeof(mark));
    memset(in,0,sizeof(in));
    vector<int> edge[MAX];
    c=0; //士兵人数计数器
    while(~scanf(" %c%c%c",&a,&x,&b)) //读入士兵A和士兵B以及他们的身高关系
    {
        it1=a-'A';
        if(mark[it1]==0)
        {
            mark[it1]=1; c++; //士兵A参与排队
        }
        it2=b-'A';
        if(mark[it2]==0)
        {
            mark[it2]=1; c++; //士兵B参与排队
        }
        if(x=='>')
        {
            in[it2]++; //比士兵it2身高高的的人数
            edge[it1].push_back(it2); //it2排在it1的后面
        }
        else
        {
            in[it1]++;//比士兵it1身高高的的人数
            edge[it2].push_back(it1);
        }
    }
    priority_queue<int,vector<int>,greater<int> > Queue; //题目要求按字典序输出，想到用到优先队列
     for(i=0;i<30;i++)
     {
         if(in[i]==0&&mark[i]==1) //找出身高最高的一位或多位士兵(没人比他们的身高高)并入队
            Queue.push(i);
     }
     n=0;
     while(!Queue.empty())
     {
         int temp=Queue.top(); //士兵依次出队，并存入ans[]数组中
         Queue.pop();
         ans[n++]=temp;
         for(i=0;i<edge[temp].size();i++) //找出比士兵temp稍矮的并入队
         {
             in[edge[temp][i]]--;
             if(in[edge[temp][i]]==0)
                Queue.push(edge[temp][i]);
         }
     }
     if(n!=c) //如果入队的人数不等于总人数说明无解
        printf("No Answer!\n");
     else
     {
         for(i=0;i<n;i++)
            printf("%c",ans[i]+'A');
         printf("\n");
     }
     return 0;
}