bzoj4562: [Haoi2016]食物链--记忆化搜索

这道题其实比较水，半个小时AC= =对于我这样的渣渣来说真是极大的鼓舞

题目大意：给出一个有向图，求入度为0的点到出度为0的点一共有多少条路

从入读为零的点进行记忆化搜索，搜到出度为零的点返回1

所有返回值加起来就是答案。

“注意单独的一种孤立生物不算一条食物链。”出题人都这么好心的说了，然而第一次交的时候还是忘了= =结果成为此题第一个WA的人hh

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define maxn 100010
using namespace std;
struct node{
    int from,to,next;
}e[maxn*2];
int head[maxn],vis[maxn],dp[maxn],ind[maxn],out[maxn];
int tot,n,m,u,v;

void insert(int u, int v){
    e[++tot].from=u;
    e[tot].to=v;
    e[tot].next=head[u];
    head[u]=tot;
}

int dfs(int x){
    if (vis[x]) return dp[x];
    vis[x]=1;
    if (out[x]==0){
        dp[x]=1;
        return 1;
    }
    int sum=0;
    for (int i=head[x]; i!=-1; i=e[i].next){
        int v=e[i].to;
        sum+=dfs(v);
    }
    dp[x]=sum;
    return sum;
}

int main(){
    scanf("%d%d", &n, &m);
    memset(ind,0,sizeof(ind));
    memset(out,0,sizeof(out));
    memset(head,-1,sizeof(head));
    tot=-1;
    for (int i=1; i<=m; i++){
        scanf("%d%d", &u, &v);
        out[u]++;
        ind[v]++;
        insert(u,v);
    }
    int ans=0;
    memset(vis,0,sizeof(vis));
    memset(dp,0,sizeof(dp));
    for (int i=1; i<=n; i++)
        if (ind[i]==0 && out[i]!=0)
            ans+=dfs(i);
    printf("%d
", ans);
    return 0;
}