大田区域赛（2017-2018） A Broadcast Stations 题解

题源：CF GYM 101667

（本文最后如果有时间可以拓展成整个比赛的题解，暂时先写 A 题的题解）

题意：给定一棵 $n$ 个节点的树，$n\le 5\times 10^3$，可以在任意若干个点上花费 $v$ 的权值，使周围距离不大于 $v$ 的点都被覆盖到，要求 $v>0$。一个点可以不花费权值，问将整个树都覆盖到最小权值和为多少。

解法：首先有一个显而易见的贪心——一个点不会被多次覆盖到。因而不会存在高的点跨过一个要投入权值的低的点，覆盖到了一个更低的点。

显然此题允许一个 $O(n^2)$ 的 DP。此题的 DP 状态设计较为精巧：记 $f_{i,j}$ 为以 $i$ 为基准，$i$ 及其子树中的点最高能覆盖到 $i$ 节点以上 $j$ 处的最小权值，且其下层都已被完全覆盖。注意此处 $j$ 可以小于零。如果记每一个点的深度为 $\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]$，那么 $f_{i,j}$ 表示了 $\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]-j$ 层以下（$>\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]-j$ 的层）的点都已经被完全覆盖的最小权值。容易发现，层数不大于 $n$，点数仅 $n$ 个，因而如果能做到 $O(1)$ 的单步转移，就可以做到 $O(n^2)$ 的复杂度。

此外，注意到——覆盖到更高处的权值只会比覆盖的更低层的权值大——毕竟覆盖范围远。因而这个 DP 的定义自带一个后缀 $\min$。即，$\forall j_1<j_2,f_{i,j_1}\le f_{i,j_2}$。

那么这个 DP 怎么转移呢？

首先，叶子节点的情况非常好处理——显然叶子节点不允许支持 $j<0$ 的情况，因而 $\forall j<0,f_{i,j}=0$。对于 $j=0$ 的情况，注意到自己不能给自己覆盖一个 $v=0$ 的权值，因而没有办法，放在一边。$\forall j>0,f_{i,j}=j$——只需要在这个节点花费权值 $j$ 即可。

对于非叶子节点，有两种情况：

1. 自己不努力，全靠儿子带。显然对于 $j\le 0$，均属于这一类。因为只要自己这里投入了权值，那么必然 $j>0$。考虑这种情况，如果从它这里只能覆盖 $\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]-j$ 层，那么对于其儿子，它们也只需要覆盖到这里即可。对应于 DP 状态，即是 $f_{so{n}_i,j+1}$。显然如果覆盖深度低于 $i$，那么不存在儿子之间互相覆盖的问题，即，一个儿子的子树内覆盖的范围超过了 $i$，进入到了别的子树。而对于 $j>0$ 的一部分情况，也是如此，但是此处就出现了刚刚的问题——子树覆盖范围超过了子树，到达了别的子树。注意到不同儿子之间的距离都是 $2$，因而如果存在某一个儿子非常努力，将覆盖范围拉到了 $\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]-j$，那么对于其他所有的儿子，它们就不需要那么努力了——因为对于其他儿子来说，$\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]+j-2$ 即更高的地方（注意这部分有一部分是在其他儿子的子树中的）都已经被这一个争气的儿子覆盖了，它们只需要把它子树里剩下的部分填完即可。
2. 自己努力往上带。这种情况最简单了，如果自己覆盖到了 $\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]-j$，那么儿子们都可以舒服一点，$\mathrm{d}\mathrm{e}\mathrm{p}\mathrm{t}\mathrm{h}[i]+j$ 以上都不用管了，只用覆盖剩下的情况。

最后取一个后缀 $\min$ 即可。

#include <cstdio>
#include <algorithm>
#include <memory.h>
using namespace std;
const int base = 5000;
struct line
{
    int from;
    int to;
    int next;
};
struct line que[10005];
int cnt, headers[5005];
void add(int from,int to)
{
    cnt++;
    que[cnt].from = from;
    que[cnt].to = to;
    que[cnt].next = headers[from];
    headers[from] = cnt;
}
int f[5005][10005];
int dfs(int place,int father)
{
    bool flag = 0;
    int maxdepth = 0;
    for (int i = headers[place];i;i=que[i].next)
        if(que[i].to!=father)
        {
            flag = 1;
            maxdepth = max(maxdepth, dfs(que[i].to, place));
        }
    if(!flag)//叶子节点
    {
        for (int k = 1; k <= base;k++)
            f[place][base + k] = k;
        for (int k = -base; k < 0;k++)
            f[place][base + k] = 0;
        return 1;
    }
    for (int k = -base; k <= base;k++)
    {
        if(k<0)//第一种情况的 j<0
        {
            f[place][base + k] = 0;
            for (int i = headers[place]; i;i=que[i].next)
                if(que[i].to!=father)
                    f[place][base + k] += f[que[i].to][base + k + 1];//注意：每个儿子都要加，因为无法相互覆盖，只能“个人自扫门前雪”了。
        }
        else
        {
            int sum = 0;
            for (int i = headers[place]; i;i=que[i].next)//预处理出 \sum f_son_i,-k，因为其他儿子都是坐享其成，只有一个儿子出力，可以预处理计算
                if(que[i].to!=father)
                    sum += f[que[i].to][base - k];
            for (int i = headers[place]; i;i=que[i].next)
                if(que[i].to!=father)
                    f[place][base + k] = min(f[place][base + k], sum - f[que[i].to][base - k] + f[que[i].to][base + k + 1]);//que[i].to 是本轮争气的那个儿子
            if(k>0)
                f[place][base + k] = min(f[place][base + k], sum + k);//第二种情况，自己努力往上带
        }
    }
    int minimum = f[place][base + base];
    for (int k = base - 1; k >= -base;k--)//取一个后缀 min。大于零的部分一定满足，把小于零的情况给处理掉。
        if (minimum > f[place][base + k])
            minimum = f[place][base + k];
        else
            f[place][k + base] = minimum;
    return maxdepth + 1;
}
int main()
{
    int n, a, b;
    memset(f, 0x01, sizeof(f));
    //最大值不可以设置的过大，否则会溢出
    scanf("%d", &n);
    for (int i = 1; i < n;i++)
    {
        scanf("%d%d", &a, &b);
        add(a, b);
        add(b, a);
    }
    dfs(1, 1);
    printf("%d", f[1][base]);
    return 0;
}