BZOJ 4562: [Haoi2016]食物链 (记忆化搜索)

原题链接：传送门

题意：给n个物种和m条能量流动关系，求其中的食物链条数。

单独的一种孤立生物不算一条食物链。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int N = 1e5+10;
vector<int> Vi[N];
int dp[N],degree[N];
int n,m;

int dfs(int x) {
    if(dp[x] != -1) {       //说明已经搜过了,直接返回其有多少链 
        return dp[x];
    }
    dp[x] = 0;      //标记并清零 
    int num = Vi[x].size();     //吃的物种数 
    for(int i=0; i<num; i++) {      //遍历食物链下一级 
        int X = Vi[x][i];
        dp[x] += dfs(X);
    }
    if(num == 0) { 
        dp[x] = 1;
    }
    return dp[x];
}

int main() {

    memset(dp,-1,sizeof(dp));
//  memset(degree,0,sizeof(degree));

//  for(int i=0; i<n; i++) {    //清空Vi
//      Vi[i].clear();
//  }

    scanf("%d%d",&n,&m);
    for(int i=0; i<m; i++) {
        int a,b;
        scanf("%d%d",&a,&b);
        degree[b] ++;           //入度标记 
        Vi[a].push_back(b);     //把b放入容器a,表示a吃b 
    }

    for(int i=0; i<n; i++) {
        if(Vi[i].size() > 0 && degree[i] == 0) {    //如果容器内有值(吃其他的),而且度为0 
            Vi[0].push_back(i);                     //说明是一个食物链的源头 
        }                                           //Vi[0]中存食物链顶端的物种 
    }

    if(Vi[0].size() != 0) {     //若食物链顶端没物种，则没有完整食物链 
        dfs(0);
        printf("%d\n",dp[0]);
    } else
        printf("0\n");

    return 0;
}