2020CCPC秦皇岛 k Kingdom’s Power


题意:有无数个军队可以从树根1出发,每花费一个点可以让军队走一步,问遍历完整个树的最小花费。
思路:先记录每个节点的高度,用节点根据它子树的高度的高度来排序,然后节点的高度为子树的最高高度+1,然后类似树形dp,从上往下递归h和与从下往上走的值取min。最后把所以的根节点的值加起来。

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int N=1000010;
typedef long long ll;
int n;
vector<pair<int,int>>v[N];
int geth(int u)
{
   
	int hh=0;
	if(!v[u].size())	return 1;
	for(int i=0;i<v[u].size();i++)
	{
   
		v[u][i].first=geth(v[u][i].second);
	}
	sort(v[u].begin(),v[u].end());
	return v[u].back().first+1;
}
int val[N];
int dfs(int u,int h,int V)
{
   
	val[u]=V;
	if(!v[u].size())	
	return 1;
	int t=V;
	for(int i=0;i<v[u].size();i++)
	{
   
		t=min(h,dfs(v[u][i].second,h+1,t+1));
	}
	return t+1;
}
int main()
{
   
	int t;
	scanf("%d",&t);
	int k=0;
	while(t--)
	{
   	
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
			v[i].clear();
		for(int i=1;i<=n;i++)
			val[i]=0;
		for(int i=2;i<=n;i++)
		{
   
			int fa;
			scanf("%d",&fa);
			v[fa].push_back({
   0,i});
		}
		geth(1);
		dfs(1,0,0);
		ll ans=0;
		for(int i=1;i<=n;i++)
			if(!v[i].size())
			{
   
				ans+=val[i];
			// cout<<val[i]<<endl;
			}
		printf("Case #%d: ",++k);
		printf("%lld\n",ans);
	}
}
全部评论

相关推荐

头像
11-27 14:28
长沙理工大学
刷算法真的是提升代码能力最快的方法吗?&nbsp;刷算法真的是提升代码能力最快的方法吗?
牛牛不会牛泪:看你想提升什么,代码能力太宽泛了,是想提升算法能力还是工程能力? 工程能力做项目找实习,算法也分数据结构算法题和深度学习之类算法
点赞 评论 收藏
分享
ProMonkey2024:5个oc?厉害! 但是有一个小问题:谁问你了?😡我的意思是,谁在意?我告诉你,根本没人问你,在我们之中0人问了你,我把所有问你的人都请来 party 了,到场人数是0个人,誰问你了?WHO ASKED?谁问汝矣?誰があなたに聞きましたか?누가 물어봤어?我爬上了珠穆朗玛峰也没找到谁问你了,我刚刚潜入了世界上最大的射电望远镜也没开到那个问你的人的盒,在找到谁问你之前我连癌症的解药都发明了出来,我开了最大距离渲染也没找到谁问你了我活在这个被辐射蹂躏了多年的破碎世界的坟墓里目睹全球核战争把人类文明毁灭也没见到谁问你了(别的帖子偷来的,现学现卖😋)
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务