BZOJ2109 NOI2010 Plane 航空管制 拓扑排序

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;

const int MAXN=2000+2;
const int MAXM=10000+2;
struct NODE{
    int u,k;
}node[MAXN];
struct HASH{
    int v;
    HASH *next;
    HASH(){}
    HASH(int _v,HASH *_next):v(_v),next(_next){}
}*table[MAXN],mem[2*MAXM];
int N,M,k[MAXN],degree[MAXN],d[MAXN],cnt,q[MAXN],s,t;

bool cmp(NODE x,NODE y){ return x.k<y.k;}

void Insert(int u,int v){ table[u]=&(mem[cnt++]=HASH(v,table[u]));}

void Topological_Sort(int u){
    memcpy(d,degree,sizeof(degree));
    s=t=0;
    for(int i=1,j=1;i<=N;i++){
        for(;j<=N && node[j].k<i;j++)
            if(!d[node[j].u] && node[j].u!=u) q[t++]=node[j].u;

        if(s<t)
            for(HASH *p=table[q[s++]];p;p=p->next){
                d[p->v]--;
                if(!d[p->v] && p->v!=u && k[p->v]<i) q[t++]=p->v;
            }
        else return;
    }
}

int main(){
    scanf("%d %d",&N,&M);
    for(int i=1;i<=N;i++){
        scanf("%d",&node[i].k);
        node[i].u=i,k[i]=node[i].k=N-node[i].k;
    }
    sort(node+1,node+N+1,cmp);
    for(int i=1,u,v;i<=M;i++){
        scanf("%d %d",&u,&v);
        Insert(v,u),degree[u]++;
    }

    Topological_Sort(-1);
    for(int i=N-1;i>=0;i--){
        printf("%d",q[i]);
        if(!i) printf("\n");
        else printf(" ");
    }

    for(int i=1;i<=N;i++){
        Topological_Sort(i);
        printf("%d ",N-t);
    }

    return 0;
}