标签：Kingdom minl Val Power dfs2 int tr 2020ccpc return

K - Kingdom’s Power

比赛的时候思路出了 但不会证。

赛后敲了就过了。

#include <bits/stdc++.h>
using namespace std;

const int N = 1000010;
#define ll long long
int n;

vector <pair<int, int>> tr[N];
int depth[N];
int val[N];

int dfs1(int u, int dep) {
    if(tr[u].empty()) return 1;

    depth[u] = dep;
    for(auto &x : tr[u])
        x.first = dfs1(x.second, dep+1);

    sort(tr[u].begin(), tr[u].end());
    return tr[u].back().first + 1;
}

int dfs2(int u, int Val) {
    val[u] = Val;
    if(tr[u].empty()) return 1;

    int minl = Val;
    for(auto &x : tr[u]) {
        minl = min(depth[u], dfs2(x.second, minl+1));
    }
    return minl + 1;
}

int main() {
    int t;
    int fa;
    scanf("%d", &t);
    for(int Case = 1;Case <= t;Case ++) {
        scanf("%d", &n);
        for(int i = 1;i <= n;i ++) tr[i].clear();
        for(int i = 2;i <= n;i ++) {
            scanf("%d", &fa);
            tr[fa].push_back(make_pair(0, i));
        }
        dfs1(1, 0);
        dfs2(1, 0);
		ll ans=0;
		for(int i=1;i<=n;i++)
			if(tr[i].empty())
				ans+=val[i];

        printf("Case #%d: %d\n", Case, ans);
    }
}

标签：Kingdom,minl,Val,Power,dfs2,int,tr,2020ccpc,return
来源： https://blog.csdn.net/binggui2/article/details/121144382

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。