首页 > 代码库 > HDU 1285 确定比赛名次(拓扑排序+优先队列)

HDU 1285 确定比赛名次(拓扑排序+优先队列)

链接:http://acm.hdu.edu.cn/showproblem.php?pid=1285

Problem Description
有N个比赛队(1<=N<=500),编号依次为1,2,3,。。。。,N进行比赛,比赛结束后,裁判委员会要将所有参赛队伍从前往后依次排名,但现在裁判委员会不能直接获得每个队的比赛成绩,只知道每场比赛的结果,即P1赢P2,用P1,P2表示,排名时P1在P2之前。现在请你编程序确定排名。
 

Input
输入有若干组,每组中的第一行为二个数N(1<=N<=500),M;其中N表示队伍的个数,M表示接着有M行的输入数据。接下来的M行数据中,每行也有两个整数P1,P2表示即P1队赢了P2队。
 

Output
给出一个符合要求的排名。输出时队伍号之间有空格,最后一名后面没有空格。

其他说明:符合条件的排名可能不是唯一的,此时要求输出时编号小的队伍在前;输入数据保证是正确的,即输入数据确保一定能有一个符合要求的排名。
 

Sample Input
4 3 1 2 2 3 4 3
 

Sample Output
1 2 4 3
 

分析:

简单的拓扑排序,唯一可能出现的地方就是题目中要求同样拓扑序的两个编号小的要在前面,这点可以通过优先队列或者堆来实现;

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#define MAXN 505
#define RST(N)memset(N, 0, sizeof(N))
using namespace std;

int n, m, u, v;
bool first;
int graph[MAXN][MAXN], degree[MAXN];  //图的邻接矩阵,每个顶点的入度;
priority_queue <int, vector<int>, greater<int> > q;   //优先队列;

void Init()
{
    RST(graph), RST(degree);    //初始化数组;
    while(!q.empty()) q.pop();   //队列清空;
    for(int i=0; i<m; i++) {
        scanf("%d %d", &u, &v);
        if(!graph[u][v]) {    //防止重边;
            graph[u][v] = 1;
            degree[v]++;
        }
    }
    first = 1;
    for(int i=1; i<=n; i++) if(!degree[i]) q.push(i);
}

int main()
{
    while(~scanf("%d %d", &n, &m)) {
        Init();
        while(!q.empty()) {
            int current = q.top();
            q.pop();
            if(first) { printf("%d", current); first = 0; }
            else printf(" %d", current);
            for(int i=1; i<=n; i++) if(graph[current][i]) {
                degree[i]--;
                if(degree[i] == 0) q.push(i);
            }
        }
        puts("");
    }
    return 0;
}