HDU1285: 确定比赛名次

题目描述

http://acm.hdu.edu.cn/showproblem.php?pid=1285
大概意思是给出比赛记录求拓扑排序
###题目分析

• 题目明确不存在有环的输入
• 有个坑是测试数据中有重边数据（不过不影响我的代码）可能需要输入时去重边
• 题目要求编号小的靠前，有的大神用了优先队列，我这里用了一个vector存所有入度为0的点，并且每一次都进行一次排序，取出队列中编号的最小的点(其实和优先队列是一样的了)
• 遇到一个小问题：我以为输出的拓扑排序中点的个数一定和输入的点个数一样，所以在输出时遍历0到n，结果因此WA了很多次，后来改为统计cnt(见代码)才AC

代码

#include <iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define N 501
using namespace std;
vector<int> edge[N];
vector<int> Q,res;
int inDegree[N];
int main() {
    freopen("in.txt","r",stdin);
    int n,m;
    while(cin>>n>>m){
        for(int i=1;i<=n;i++){
            inDegree[i]=0;
            edge[i].clear();
        }
        int a,b;
        while(m--){
            cin>>a>>b;
            inDegree[b]++;
            edge[a].push_back(b);
        }
        for(int i=1;i<=n;i++){
            if(inDegree[i]==0){
                Q.push_back(i);
            }
        }
        int cnt=0;
        while(!Q.empty()){
            sort(Q.begin(),Q.end());//将队列中的点按大小排序（题目要求） 
            int newP=Q.front();
            res.push_back(newP);
            Q.erase(Q.begin());//把第一个删掉 
            cnt++;
            for(int i=0;i<edge[newP].size();i++){
                inDegree[edge[newP][i]]--;
                if(inDegree[edge[newP][i]]==0){
                    Q.push_back(edge[newP][i]);
                }
            }
        }
        //之前一直WA就是因为cnt，我直接用n,改成cnt后AC了，这点我还是有点疑惑的 
        for(int i=0;i<cnt-1;i++){
            cout<<res[i]<<" ";
        }
        cout<<res[cnt-1]<<endl;
        res.clear();
    } 
    return 0;
}