codeforce 图论dfs+贪心

题目链接:https://codeforces.ml/contest/1337/problem/C

大意:给出n,k和一个点数量为n的树,让其中k个结点变为工业城市,其余为旅游城市。而每个工业城市到根节点1的路径上存在的旅游城市数量之和求最大,并输出最大值。

样例:
Examples
inputCopy
7 4
1 2
1 3
1 4
3 5
3 6
4 7
outputCopy
7
inputCopy
4 1
1 2
1 3
2 4
outputCopy
2
inputCopy
8 5
7 5
1 7
6 1
3 7
8 3
2 1
4 5
outputCopy
9

解题报告:这题我当时没做出来,考虑到了该选最深的节点,发现策略是错误的,如果选了他的父节点就会影响到他,因为题目要算的是风景区的总和最大,看了网上的题解,就是把他到根节点的距离-他的子树的个数,sort一下选前k大。加起来就是答案,因为求的是答案总和,如果在底下最深的节点前面选,必然会让每个节点的答案都减1,所以可以减到父节点上。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#define IL inline
#define x first
#define y second
typedef long long ll;
using namespace std;
const	int N=2e5+10;
int cnt[N],d[N];
vector<int>v[N];
void dfs(int u,int pre)
{
   
	cnt[u]=1;
	
	for(int i=0;i<v[u].size();i++)
	{
   
		int j=v[u][i];
		if(j==pre)	continue;
		d[j]=d[u]+1;
		dfs(j,u);
		cnt[u]+=cnt[j];		
	}
}
int main()
{
   
	int n,k;
	cin>>n>>k;
	for(int i=0;i<n-1;i++)
	{
   
		int a,b;
		cin>>a>>b;
		v[a].push_back(b);
		v[b].push_back(a);
		}
		d[1]=1;
	dfs(1,0);
	for(int i=1;i<=n;i++)
	d[i]-=cnt[i];
	sort(d+1,d+n+1);
	reverse(d+1,d+n+1);
	ll res=0;
	for(int i=1;i<=k;i++)
	res+=d[i];
	cout<<res<<endl;
    return 0;
}




全部评论

相关推荐

02-10 13:41
西南大学 Java
点赞 评论 收藏
分享
HR_丸山彩同学:你的项目描述里,系统设计讲了很多:MemCube是什么、三级存储架构怎么设计、四种遗忘策略分别是什么。这些面试的时候讲没问题,但简历上不需要这么细。 简历要突出的是影响力,不是实现细节。面试官看简历的时候想知道的是「这个项目有多大价值」,不是「这个项目具体怎么实现的」。实现细节是面试时候聊的 怎么改:技术细节可以精简为一句「采用三级存储架构+四种遗忘策略」,把省出来的篇幅用来写影响力。比如:项目有没有开源?有没有写成技术博客?有没有被别人使用过? 校园经历没有任何信息量,任何人都可以写这句话,写了等于没写。更关键的是,你投的是技术岗,校园活动经历本来就不是加分项。如果非要写,必须写出具体的数字和成果。如果你没有这些数字,那就老老实实删掉 「端到端耗时缩减30-40%」要给出确切数字和绝对值。从1000ms降到600ms是降了40%,从100ms降到60ms也是降了40%,但这两个含义完全不一样。其他也是,涉及到数据,准备好证据,口径统一,面试会问 「熟练」「熟悉」「了解」混在一起用,读起来很乱。而且「了解前端需求」最好改成「具备前后端协作经验」
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务