题目链接: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;
}