洛谷P3183 [HAOI2016]食物链

洛谷P3183 [HAOI2016]食物链
记忆化搜索

#include <bits/stdc++.h> 
#define For(i,j,k) for(int i=j;i<=k;i++) 
using namespace std ; 

const int N = 100011,M = 200011 ; 
int n,m,cnt,ans ; 
int f[N],head[N],IN[N],OUT[N] ; 
struct node{
    int to,pre ; 
}e[M] ; 

inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline void add(int x,int y) 
{
    e[++cnt].to = y ; 
    e[cnt].pre = head[x] ; 
    head[x] = cnt ; 
}

inline int dfs(int u) 
{
    if(f[u]) return f[u] ; 
    if(IN[u]&&!OUT[u]) return f[u] = 1 ;          //  单个生物是无法成为食物链的 
    for(int i=head[u];i;i=e[i].pre) {
        int v = e[i].to ;  
        f[u]+=dfs(v) ;  
    }
    return f[u] ; 
}

int main() 
{
    n = read() ; m = read() ; 
    int x,y ; 
    while(m--) {
        x = read() ; y = read() ; 
        add(x,y) ; OUT[x]++ ; IN[y]++ ; 
    }
    For(i,1,n) if(!f[i]) ans+=dfs(i) ; 
    printf("%d
",ans) ; 
    return 0 ; 
}