大意:
有一个有 n 个点的树,以 1 为根,你可以选择 k 个节点,使得这 k 个节点到 1 节点的最短路径中经过的非选择的点最多。
思路:
简单树形dp
我们肯定会优先最好的叶子结点,这个能保证中间路径最长,相同路径下选叶子节点比选非叶子节点一定是要优的。所以如果我们选了一个非叶子节点,代表它的所有儿子我们肯定都选过了,那么就得把对应的重复贡献减掉。这样一综合就得到了对应的公式:每一个点选上之后对答案的贡献应该是它的深度减去以它为根的子树的大小。
这里要求每一个点的深度以及子树大小,一遍dfs就可以直接带走
水水更健康~
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=2e5+10;
#define endl '\n'
ll n,k,a,b;
struct ty
{
ll t,next;
}edge[N<<1];
ll head[N];
ll cnt=0;
void add(ll a,ll b)
{
edge[++cnt].t=b;
edge[cnt].next=head[a];
head[a]=cnt;
}
struct sds{
ll dep,son;//深度,子树大小
}ans[N];
bool cmp(sds a,sds b)
{
return a.dep-a.son>b.dep-b.son;
}
void dfs(ll u,ll p,ll dep)
{
ans[u].dep=dep;
ans[u].son=1;
for(int i=head[u];i!=-1;i=edge[i].next)
{
ll y=edge[i].t;
if(y==p) continue;
dfs(y,u,dep+1);
ans[u].son+=ans[y].son;
}
}
int main()
{
cin>>n>>k;
memset(head,-1,sizeof head);
for(int i=1;i<n;++i)
{
cin>>a>>b;
add(a,b);
add(b,a);
}
dfs(1,0,1);
sort(ans+1,ans+1+n,cmp);
ll sd=0;
for(int i=1;i<=k;++i)
{
sd+=ans[i].dep-ans[i].son;
}
cout<<sd<<endl;
return 0;
}