洛谷 P3479 [POI2009]GAS-Fire Extinguishers（树形DP）

题目链接：洛谷 P3479

题目大意：在一棵n个节点的树上放置灭火器，每个灭火器可以覆盖与其所在节点距离不超过k的节点，每个灭火器最多能覆盖s个节点，求至少需要多少灭火器可以使得所有节点都被覆盖。（n<=100000，s<=n，k<=20）

题解：树形DP，设f[i][j]表示以i为根的子树里已经放置的与i的距离为j的灭火器还可以覆盖的点数，g[i][j]表示以i为根的子树里与i距离为j的未被覆盖的点数。转移的时候容易想到的是用f[i][j]覆盖g[i][k-j]和当g[i][k]>0时要在i放置灭火器。但是还要注意，如果两个节点距离i分别为j和k-j-1，有可能i就是两个节点的lca，也需要在此时考虑是否用其中一个节点的灭火器覆盖另一个节点，否则在深度更小处就处理不到这种情况了。更新根节点的时候也要特殊处理一下（详见代码）。

总结：涉及的树上的路径要留意LCA处、根节点处是否需要特殊考虑。

code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 100005
using namespace std;
inline int read()
{
    char c=getchar(); int num=0,f=1;
    while (c<'0'||c>'9') { if (c=='-') f=-1; c=getchar(); }
    while (c<='9'&&c>='0') { num=num*10+c-'0'; c=getchar(); }
    return num*f;
}
struct edge{
    int to,ne;
}e[N<<1];
int n,s,k,tot,head[N],f[N][25],g[N][25],ans;
void push(int x,int y)
{
    e[++tot].to=y; e[tot].ne=head[x]; head[x]=tot;
    e[++tot].to=x; e[tot].ne=head[y]; head[y]=tot;
}
void dfs(int now,int pre)
{
    for (int i=head[now];i;i=e[i].ne)
    {
        int v=e[i].to; if (v==pre) continue;
        dfs(v,now);
        for (int j=1;j<=k;j++)
         f[now][j]=min(n,f[now][j]+f[v][j-1]),g[now][j]+=g[v][j-1];
    }
    g[now][0]++;
    if (g[now][k])
    {
        int t=(g[now][k]>0)?((g[now][k]-1)/s+1):0;
        f[now][0]=min(1ll*n,1ll*t*s),ans+=t;
    }
    for (int i=0;i<=k;i++)
    {
        int j=k-i,d=min(f[now][i],g[now][j]);
        f[now][i]-=d; g[now][j]-=d;
    }
    for (int i=0;i<k;i++)  //注意
    {
        int j=k-i-1,d=min(f[now][i],g[now][j]);
        f[now][i]-=d; g[now][j]-=d;
    }
}
int main()
{
    n=read(); s=read(); k=read();
    for (int i=1;i<n;i++) push(read(),read());
    dfs(1,0);
    for (int i=0;i<=k;i++)
     for (int j=k;j>=0;j--) if (i+j<=k)  //注意
     {
         int d=min(f[1][i],g[1][j]);
         f[1][i]-=d,g[1][j]-=d;
     }
    tot=0;
    for (int i=0;i<=k;i++) tot+=g[1][i];
    if (tot>0) tot=(tot-1)/s+1;
    printf("%d",ans+tot);
    return 0;
}