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\n", ans);
    return 0;
}