其实总的来说就是不会做….QAQ
我们对于每一个节点记录两个值,need[i][j]代表需要被i节点覆盖的距离为j的节点数目,cover[i][j]代表可以被i节点覆盖的距离为j的节点数目…
然后如果一个节点的need大于cover那么就贪心的向这里放一些救火站…
相邻节点的need值和cover值可以互相抵消…但是我们贪心的只让距离为i和距离为i-1的两值相消…因为如果相消的两项距离相差太大就是浪费…
代码如下:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using namespace std;
const int maxn=100000+5,maxk=20+5;
int n,s,k,hd[maxn],to[maxn*2],nxt[maxn*2],cnt;
long long cover[maxn][maxk],need[maxn][maxk],ans;
inline void add(int x,int y){
to[cnt]=y;nxt[cnt]=hd[x];hd[x]=cnt++;
}
inline void modify(int root,int lala){
if(lala==0){
for(int i=0;i<=k;i++)
for(int j=i;j>=i-1&&cover[root][i];j--){
long long tmp=min(cover[root][i],need[root][j]);
cover[root][i]-=tmp,need[root][j]-=tmp;
}
}
else{
for(int i=0;i<=k;i++)
for(int j=i;j>=0&&cover[root][i];j--){
long long tmp=min(cover[root][i],need[root][j]);
cover[root][i]-=tmp,need[root][j]-=tmp;
}
}
}
inline void dfs(int root,int fa){
need[root][0]=1;
for(int i=hd[root];i!=-1;i=nxt[i])
if(to[i]!=fa){
dfs(to[i],root);
for(int j=0;j<k;j++)
need[root][j+1]+=need[to[i]][j];
for(int j=0;j<k;j++)
cover[root][j]+=cover[to[i]][j+1];
}
long long rema=need[root][k]-cover[root][k];
if(rema>0)
cover[root][k]+=(rema+s-1)/s*s,ans+=(rema+s-1)/s;
modify(root,0);
}
signed main(void){
scanf("%d%d%d",&n,&s,&k);cnt=0;
memset(hd,-1,sizeof(hd));
for(int i=1,x,y;i<n;i++)
scanf("%d%d",&x,&y),add(x,y),add(y,x);
dfs(1,0);modify(1,1);long long tmp=0;
for(int i=0;i<=k;i++)
tmp+=need[1][i];
ans+=(tmp+s-1)/s;
printf("%d\n",ans);
return 0;
}
by >_< NeighThorn