珂朵莉的值域连续段（树的dfs）

念兮兮

已于 2022-10-07 13:39:48 修改

阅读量150

点赞数

分类专栏：每日练习文章标签：深度优先 c++ 算法图论

于 2022-10-05 19:03:40 首次发布

本文链接：https://blog.csdn.net/qq_52081267/article/details/127175520

版权

每日练习专栏收录该内容

18 篇文章 0 订阅

订阅专栏

原题链接

解题思路：本题我们可以先找到树的根节点，然后进行一次树遍历，同时记录每颗子树的节点个数、节点最大值和节点最小值，判断数在值域上连续可以用节点最大值减去节点最小值加1是否等于节点个数来判断

C++代码

#include <bits/stdc++.h>
// #define int long long 
using namespace std;
const int N = 1e5 + 10, M = N;
int h[N], e[M], ne[M], idx;
int d[N], maxn[N], minn[N], cnt[N];
int root, ans, n;

void add(int a, int b)  // 添加一条边a->b
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}


void dfs(int u, int fa)
{
    cnt[u] = 1;
    maxn[u] = u;
    minn[u] = u;
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == fa) continue;
        dfs(j, u);
        maxn[u] = max(maxn[u], maxn[j]);
        minn[u] = min(minn[u], minn[j]);
        cnt[u] += cnt[j];
    }
    if (cnt[u] == maxn[u] - minn[u] + 1)
    {
        ans ++ ;
    }
}

signed main()
{
    memset(h, -1, sizeof h);
    cin >> n;
    int a, b;
    for (int i = 1; i < n; i ++ )
    {
        cin >> a >> b;
        add(a, b), d[b] ++ ;
    }
    for (int i = 1; i <= n; i ++ )
    {
        if (d[i] == 0)
        {
            root = i;
            break;
        }
    }
    dfs(root, -1);
    cout << ans << endl;
    return 0;
}