Codeforces Round #635 (Div. 1)A. Linova and Kingdom

Codeforces Round #635 (Div. 1)A. Linova and Kingdom

题目大意：

给你一个含有 n 个节点的树，1号节点为根节点，要求选定 k 个节点作为黑点，目标为 所有黑点到根节点的最短路径上经过的白点数之和。

思路：

• 如果一个点为黑点，那么这棵子树全为黑点。

  • 证：父节点u，后代节点v,分别产生贡献为，dep_u < dep_v,故当u为黑时，v为黑。
• 每个黑点的贡献为 dep[ i ] — {到根节点路径上的黑点数} = 》因为每个黑点会因为每个后代为黑点而产生 -1 的贡献，故每个黑点可以用 dep[i] — sz[i] 替代。
• 要求贡献最大，故对dep[i] — sz[i] 排序后贪心即可。

input:

7 4
1 2
1 3
1 4
3 5
3 6
4 7

output:

7

code:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define Please return 
#define Accepted 0
#define int long long
#define endl "\n"
typedef long long LL;
typedef pair<int,int> PII;
typedef pair<double, double> PDD;
const int N = 200010,M=2*N,INF=0x3f3f3f3f;
int n,k;
vector<int>e[N];
struct node{
	int sz,dep,id;
};
bool cmp(node x,node y){ 
	return x.dep-x.sz>y.dep-y.sz;
}
int dp[N],dep[N]; 
void dfs(int u,int fa,int d)
{
	dp[u]=1;dep[u]=d;
	for(auto v:e[u])
	{
		if(v==fa)	continue;
		dfs(v,u,d+1);
		dp[u]+=dp[v];
	}
} 
void slove(int _case)
{
	cin>>n>>k;
	for(int i=1;i<n;i++){
		int u,v;cin>>u>>v;
		e[u].push_back(v),e[v].push_back(u);
	}
	dfs(1,-1,1);
	int ans=0;
	vector<node> a;
	for(int i=1;i<=n;i++)	a.push_back({dp[i],dep[i],i});
	sort(a.begin(),a.end(),cmp);  
	for(int i=0;i<k;i++){
		ans+=a[i].dep-a[i].sz;
	}
	cout<<ans<<endl;
}

signed main()
{
	ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int T=1;
//	cin>>T;
	for(int _case=1;_case<=T;_case++)
	{
		slove(_case);
	}
	Please Accepted;
}