2的指数,前n-1个数的和不如第n个数大。所以我们要把n-k个最大的数保留下来。做法就是贪心排个序,每次找一下还能路径上经过的点数不超过还能放在集合里点数的最大的点。咋找路径呀,那就用树上倍增。
#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
vector<int> vec[1000010];
int p[1000010][20],n,k;
bool vis[1000010];
void dfs(int u,int fa)
{
p[u][0]=fa;
for (int i=1; i<20; i++)
{
p[u][i]=p[p[u][i-1]][i-1];
}
for (int i=0; i<vec[u].size(); i++)
{
int v=vec[u][i];
if (v==fa) continue;
dfs(v,u);
}
}
int main()
{
scanf("%d%d",&n,&k);
memset(vis,false,sizeof(vis));
for (int i=1; i<n; i++)
{
int a,b;
scanf("%d%d",&a,&b);
vec[a].push_back(b);
vec[b].push_back(a);
}
memset(p,0,sizeof(p));
dfs(n,n);
int on=n-1-k;
vis[n]=true;
for (int i=n-1; i>0; i--)
{
if (vis[i]) continue;
int tmp=i;
int sum=0;
for (int j=19; j>=0; j--)
{
if (!vis[p[tmp][j]])
{
tmp=p[tmp][j];
sum+=(1<<j);
}
}
sum++;
if (sum<=on)
{
on-=sum;
tmp=i;
while (!vis[tmp])
{
vis[tmp]=true;
tmp=p[tmp][0];
}
}
}
for (int i=1; i<=n; i++)
{
//cout<<"i="<<i<<" vis="<<vis[i]<<endl;
if (!vis[i]) printf("%d ",i);
}
printf("\n");
return 0;
}