BZOJ1117: [POI2009]救火站Gas

考虑贪心的去放消防站，让每个消防站的层数尽量浅，从而覆盖子树里的点的同时能对子树外产生贡献

那么x要放消防站的时候，一定是x的子树里还未被覆盖的最远的点距离x的距离为k，如果这样的点有c个，我们要放$\lceil \dfrac{c}{s} \rceil$个消防站，这些消防站覆盖完了这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;
}