bzoj-4562：[Haoi2016]食物链

最新推荐文章于 2020-12-01 19:04:52 发布

ZiMoXuan_

最新推荐文章于 2020-12-01 19:04:52 发布

阅读量245

点赞数

分类专栏：深搜文章标签：记忆化搜索

本文链接：https://blog.csdn.net/ZiMoXuan_/article/details/80426449

版权

深搜专栏收录该内容

1 篇文章 0 订阅

订阅专栏

食物链

Time Limit: 10 Sec Memory Limit: 128 MB
题目描述:

现在给你n个物种和m条能量流动关系，求其中的食物链条数。

物种的名称为从1到n的编号。

m 条能量流动关系形如

a1 b1

a2 b2

a3 b3

… …

am−1 bm−1

am bm

其中 ai bi表示能量从物种ai流向物种bi。

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

输入描述:

多组数据
每组第一行两个整数n和m,接下来m行每行两个整数ai bi描述m条能量流动关系。（保证输入数据符合生物学特点，且不会有重复的能量流动关系出现）

输出描述:

每组一个整数一行，即食物网中的食物链条数。

样例输入:

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≤N≤100000,0≤m≤200000

保证答案不会超过 int 的最大值

普通深搜会超时。
由于从某个点出发的条数是固定的，我们可以搜索一遍把它记录下来，下一次直接访问记录的值。基本的记忆化搜索。

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
vector<int> a[100010];
int v[100010], vv[100010];
int dfs(int x) {
    //cout<<x<<endl;
    if (a[x].size() == 0) {
        return 1;
    }
    int ans = 0;
    for (vector<int>::iterator i = a[x].begin(); i != a[x].end(); i++) {
        //cout<<*i<<endl;
        if (vv[*i] == 0)
            vv[*i] = dfs(*i);
        ans += vv[*i];
    }
    return ans;
}
int main() {
    int n, m, x, y;
    while (~scanf("%d%d", &n, &m)) {
        memset(v, 0, sizeof(v));
        memset(vv, 0, sizeof(vv));
        for (int i = 0; i <= n; i++)
            a[i].clear();
        for (int i = 0; i < m; i++) {
            scanf("%d%d", &x, &y);
            a[x].push_back(y);
            v[y] = 1;
        }
        int ans = 0;
        for (int i = 1; i <= n; i++) {
            if (v[i] == 0 && a[i].size()) //入度为0且出度不为0
                ans += dfs(i);
        }
        printf("%d\n", ans);
    }

    return 0;
}