考虑贪心的去放消防站,让每个消防站的层数尽量浅,从而覆盖子树里的点的同时能对子树外产生贡献
那么x要放消防站的时候,一定是x的子树里还未被覆盖的最远的点距离x的距离为k,如果这样的点有c个,我们要放
⌈cs⌉
个消防站,这些消防站覆盖完了这c个点后,再去覆盖距离x距离为k-1的点,都覆盖完了后还能覆盖的点数上传给父亲
为什么这样是最优的,因为此时距离x为k,k-1的点已经全部覆盖,剩下的点和x距离<=k-2,就算我们把这些点上传给了父亲,父亲仍能覆盖这个子树剩余的最远点,同时也可以去覆盖其他子树,显然这样做一定不会更劣
因为k<=20,用num[x][j]记录x子树里还没被覆盖的点中,距离x为j的点有多少个,c[x][j]表示x能覆盖距离x为j的多少个点
每个点x用c[x][j]尝试去更新num[x][j],num[x][j-1],剩余的上传给父亲
注意c数组可能爆int qwq
code:
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 110000;
const int maxk = 22;
int n,s,K;
int ans;
int num[maxn][maxk];
ll c[maxn][maxk];
struct edge{int y,nex;}a[maxn<<1]; int len,fir[maxn];
inline void ins(const int x,const int y){a[++len]=(edge){y,fir[x]};fir[x]=len;}
int dep[maxn];
void cal(const int x,const int fa)
{
num[x][0]++;
for(int k=fir[x],y=a[k].y;k;k=a[k].nex,y=a[k].y) if(y!=fa)
{
dep[y]=dep[x]+1; cal(y,x);
for(int j=1;j<=K;j++) c[x][j-1]+=c[y][j];
for(int j=0;j<K;j++) num[x][j+1]+=num[y][j];
}
if(num[x][K])
{
int cc=(num[x][K]+s-1)/s; ans+=cc;
c[x][K]+=(ll)cc*s;
}
for(int j=0;j<=K;j++) if(c[x][j])
{
ll &tmp=c[x][j];
if(num[x][j])
{
if(num[x][j]<tmp) tmp-=num[x][j],num[x][j]=0;
else num[x][j]-=tmp,tmp=0;
}
if(tmp&&j-1>=0&&num[x][j-1])
{
if(num[x][j-1]<tmp) tmp-=num[x][j-1],num[x][j-1]=0;
else num[x][j-1]-=tmp,tmp=0;
}
}
}
int main()
{
scanf("%d%d%d",&n,&s,&K);
for(int i=1;i<n;i++)
{
int x,y; scanf("%d%d",&x,&y);
ins(x,y); ins(y,x);
}
dep[1]=1; cal(1,0);
for(int i=0;i<=K;i++)
{
ll &tmp=c[1][i];
for(int j=0;j<=i&&tmp;j++) if(num[1][j])
{
int &y=num[1][j];
if(tmp>y) tmp-=y,y=0;
else y-=tmp,tmp=0;
}
}
int oth=0; for(int i=0;i<=K;i++) oth+=num[1][i];
ans+=(oth+s-1)/s;
printf("%d\n",ans);
return 0;
}