洛谷 P3183 [HAOI2016]食物链(dfs+记忆化搜索)

https://www.luogu.com.cn/problem/P3183

题目大意:

给定n个节点,标号分别为1——n,然后给出m条有向边,问我们不同的食物链路径有多少?
输入 #1 
10 16
1 2
1 4
1 10
2 3
2 5
4 3
4 5
4 8
6 5
7 6
7 9
8 5
9 8
10 6
10 7
10 9
输出 #1 
9
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<LL,LL> PII;
const LL MAXN=1e18;
const LL N=500200,M=2002;
LL n,m;
LL mp[N];
vector<LL> g[N];
bool st[N];
int dfs(int idx)
{
    if(g[idx].size()==0) return 1;//它自己是最小的,返回当前这一个的数量
    if(mp[idx]) return mp[idx];//已经被标记过的了,直接返回
    int res=0;
    for(int i=0;i<g[idx].size();i++)
    {
        res+=dfs(g[idx][i]);//祖先辈中不同的数字下就有多少中不同的情况
    }
    return mp[idx]=res;//当前数量
}
int main()
{
    cin.tie(0); cout.tie(0); ios::sync_with_stdio(false);
    LL T=1;
    //cin>>T;
    while(T--)
    {
        cin>>n>>m;
        for(int i=1;i<=m;i++)
        {
            LL u,v;
            cin>>u>>v;
            g[v].push_back(u);//装的是自己的上级
            st[u]=true;//如果不是食物链底端,则为true
        }
        LL ans=0;
        for(int i=1;i<=n;i++)
        {
            //从低到高寻找
            //是食物链的低端并且头上有人
            if(!st[i]&&g[i].size()) ans+=dfs(i);
        }
        cout<<ans<<endl;
    }
    return 0;
}
posted @ 2022-11-01 17:50  Vijurria  阅读(52)  评论(0编辑  收藏  举报