GYM 554468 C 2022 CCPC 绵阳

GYM 554468 C 2022 CCPC 绵阳

题目大意：

给你一颗有根树，树上的每个节点都有蝴蝶，并且蝴蝶每一秒钟往根节点方向移动一次，到达根节点时消失。你可以在每一秒钟移动到某一个节点任意次，并抓住在该节点的所有蝴蝶。求抓到所有蝴蝶最大的移动次数。

基本思路：

最贪心的方案实在根节点的子节点抓蝴蝶，每一个节点抓蝴蝶的次数是以该节点为根节点的子树最大深度。

代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long int
#define Paddi ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
const int N = 1e5 + 10;
vector<int> e[N];
int dfs(int s, int fa, int dep)
{
    int ans = dep;
    for (auto i : e[s])
    {
        if (i == fa)
            continue;
        else
            ans = max(dfs(i, s, dep + 1), ans);
    }
    return ans;
}
signed main()
{
    Paddi;
    int n;
    cin >> n;
    for (int i = 1; i <= n - 1; i++)
    {
        int x, y;
        cin >> x >> y;
        e[x].push_back(y), e[y].push_back(x);
    }
    int ans = 0;
    for (auto i : e[1])
        ans += dfs(i, 1, 1);
    cout << ans << endl;

    return 0;
}