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