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;
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值