CF1336A Linova and Kingdom

题目

CF1336A Linova and Kingdom

分析

首先很容易想到可以贪心，每次一定先选深度更深的。

于是我们直接 \(dfs\) 一下再排个序取前k大的点就行了。

然后发现假了。

冷静思考一下，发现对于不是叶子的节点如果我们选了会导致我们的贡献减少，因为会“站位”，而且在选这个节点之前其叶子节点必然会先被选完。（因为“选当前节点”一定比“选当前节点的子树内节点”要劣。）

于是想到不是按照 \(dep\) 来排序而是按照 \(dep-siz\) 来排序。

然后贪心取就完了，注意开 \(ll\) 。（vp时想到第二个后还wa了我直接原地懵逼，然后发现没开ll，哈哈。）

代码

#include<bits/stdc++.h>
using namespace std;
//#ifdef ONLINE_JUDGE
//	#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
//	char buf[1<<21],*p1=buf,*p2=buf;
//#endif
template<typename T>
inline void read(T &x){
	x=0;bool f=false;char ch=getchar();
	while(!isdigit(ch)){f|=ch=='-';ch=getchar();}
	while(isdigit(ch)){x=x*10+(ch^48);ch=getchar();}
	x=f?-x:x;
	return ;
}
template<typename T>
inline void write(T x){
	if(x<0) x=-x,putchar('-');
	if(x>9) write(x/10);
	putchar(x%10^48);
	return ;
}
#define ll long long
#define ull unsigned long long
#define ld long double
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define pc putchar
#define PII pair<int,int>
#define rep(i,x,y) for(register int i=(x);i<=(y);i++)
#define dep(i,y,x) for(register int i=(y);i>=(x);i--)
const int MOD=1e9+7;
inline int inc(int x,int y){x+=y;return x>=MOD?x-MOD:x;}
inline int dec(int x,int y){x-=y;return x<0?x+MOD:x;}
inline void incc(int &x,int y){x+=y;if(x>=MOD) x-=MOD;}
inline void decc(int &x,int y){x-=y;if(x<0) x+=MOD;}
inline void chkmin(int &x,int y){if(y<x) x=y;}
inline void chkmax(int &x,int y){if(y>x) x=y;}
const int N=1e6+5,M=2e5+5,INF=1e9+7,V=3e5;
int n,m,x,a[N];
int head[N],to[N<<1],nex[N<<1],idx;
bool vis[N];
int siz[N],dep[N],dep1[N];
ll Ans;
inline void add(int u,int v){nex[++idx]=head[u];to[idx]=v;head[u]=idx;return ;}
void dfs(int x,int f){
	siz[x]=1;dep[x]=dep[f]+1;
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(y==f) continue;
		dfs(y,x);siz[x]+=siz[y];
	}
	return ;
}
void dfs1(int x,int f){
	dep1[x]=dep1[f]+(vis[x]);
	for(int i=head[x];i;i=nex[i]){
		int y=to[i];
		if(y==f) continue;
		dfs1(y,x);siz[x]+=siz[y];
	}
	if(vis[x]) Ans+=dep[x]-dep1[x];
	return ;
}
inline bool cmp(const int &x,const int &y){return dep[x]-siz[x]>dep[y]-siz[y];}
signed main(){
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
//	ios::sync_with_stdio(false);
//	double ST=clock();
	read(n),read(m);int u,v;
	rep(i,1,n) a[i]=i;
	rep(i,1,n-1) read(u),read(v),add(u,v),add(v,u);
	dfs(1,0);
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=m;i++) vis[a[i]]=true;
	dfs1(1,0);
	write(Ans);
//#ifndef ONLINE_JUDGE
//	cerr<<"\nTime:"<<(clock()-ST)/CLOCKS_PER_SEC<<"s\n";
//#endif
	return 0;
}
/*
7 6
1 2
1 3
1 4
3 5
3 6
4 7
*/