7-1 任务拓扑排序

作者 朱允刚

单位 吉林大学

一个工程被分解成n个子任务，编号为0至n-1。要完成整个工程需要完成所有的子任务。其中一些子任务必须先于另外一些子任务被完成。给定各子任务之间的先后关系，请编写程序给出一个合理的任务完成顺序，若工程不可行，程序亦能识别。

输入格式:

输入第一行为两个整数n和e，均不超过100。n表示子任务数。接下来e行，表示已知的两个子任务间的先后关系，每行为两个整数a和b，表示任务a必须先于任务b完成。

输出格式:

若工程不可行（一些子任务以自己为先决条件），输出“unworkable project”；若工程可行，输出为1行整数，每个整数后一个空格，为n个子任务的编号，表示子任务的完成顺序，如果有多种可能的顺序，则输出字典序最小者。

注：字典序，即对象在字典中的顺序。对于两个数字序列，从第一个数字开始比较，当某一个位置的数字不同时，该位置数字较小的序列，字典序较小，例如1 2 3 9比1 2 4 5小，1 2 8 9比1 2 10 3小。

输入样例1:

3 2
0 1
1 2

输出样例1:

0 1 2 

输入样例2:

3 3
0 1
1 2
2 0

输出样例2:

unworkable project

代码长度限制

16 KB

时间限制

50 ms

内存限制

64 MB

#include <iostream>
#include <queue>
#include <cstring>

using namespace std;

const int N = 1e5+10;
int e[N],ne[N],idx;//邻接表
int h[N];
int n,m;
int d[N];//各点的入度
priority_queue<int, vector<int>, greater<int>> q;//小根堆
int res[N],o;

//用邻接表储存
void add(int a,int b){
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
//拓扑排序
void topsort(){
    for(int i=0;i<n;i++){
        if(d[i]==0)q.push(i);
    }
    while(!q.empty()){
        int a = q.top();
        q.pop();
        res[o++]=a;
        for(int i=h[a];i!=-1;i=ne[i]){
            int b=e[i];
            d[b]--;
            if(d[b]==0)q.push(b);
        }
    }
    if(o==n){
       for(int i=0;i<n;i++){
           cout<<res[i]<<' ';
       }
    }
    else cout<<"unworkable project";
}
int main (){
    
    cin>>n>>m;
    //初始化
    memset(h,-1,sizeof h);
    while(m--){
        int a,b;
        cin>>a>>b;
        d[b]++;
        add(a,b);
        
    }
    topsort();
    return 0;
}