洛谷 P4017 最大食物链计数 (拓扑排序)

这是一个很好的可以锻炼思维的拓扑排序

#include
#include
#include
#include
#define MAXN 5010
#define MAXM 500010
#define M 80112002
using namespace std;
int n,m,u,v,in[MAXN],out[MAXN],head[MAXN],res,f[MAXN],cnt;
queue<int> q;
struct Edge{
    int to,nxt;
}edge[MAXM];
void addedge(int u,int v){
    edge[cnt].to=v;edge[cnt].nxt=head[u];head[u]=cnt++;
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v);out[u]++;in[v]++;
    }
    for(int i=1;i<=n;i++)if(!in[i])f[i]=1,q.push(i);
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=head[u];i!=-1;i=edge[i].nxt){
            int v=edge[i].to;
            f[v]=(f[u]+f[v])%M;in[v]--;
            if(!in[v]){
                if(!out[v])res=(res+f[v])%M;
                q.push(v);
            }
        }
    }
    printf("%d\n",res);
    return 0;
}

你可能感兴趣的:(拓扑排序)