[洛谷P4017]最大食物链计数

传送门
dp思路非常好想,就是简单的加法原理。但是状态维护的顺序必须按照拓扑序,那么我们就边拓扑排序边更新dp数组就行了。
比如i生物能吃a、b、c三种生物,那么dp[i]=dp[a]+dp[b]+dp[c]
但在此之前我们必须已经求出dp[a,b,c]的值,也就是状态维护应该按照拓扑序。
代码

#include<iostream>
#include<cstdio>
#include<queue>
#define maxn1 500005
#define maxn2 5005
#define modn 80112002
using namespace std;
struct graph{
    int nex, to;
} edge[maxn1];
int head[maxn2], tot;
inline void insert(int from,int to){
    edge[++tot].nex = head[from];
    head[from] = tot;
    edge[tot].to = to;
}
int N, M;
int dp[maxn2],in[maxn2],out[maxn2],ans;
void topo_dp(){
    queue<int> que;
    for (int i = 1; i <= N;i++){
        if(in[i]==0){
            dp[i] = 1;
            que.push(i);
        }
    }
    while(!que.empty()){
        int v = que.front();
        que.pop();
        for (int i = head[v]; i;i=edge[i].nex){
            int to = edge[i].to;
            in[to]--;
            dp[to] += dp[v];
            dp[to] %= modn;
            if(in[to]==0){
                if(out[to]==0){
                    ans += dp[to];
                    ans %= modn;
                }else
                    que.push(to);
            }
        }
    }
}
int main(){
    scanf("%d%d", &N, &M);
    int u, v;
    for (int i = 1; i <= M;i++){
        scanf("%d%d", &u, &v);
        insert(u, v);
        in[v]++;
        out[u]++;
    }
    topo_dp();
    printf("%d", ans);
}
posted @ 2021-01-07 09:09  溡沭  阅读(70)  评论(0编辑  收藏  举报