[BZOJ4557][JLoi2016]侦察守卫（树形DP）

把原图视为有根树，以$1$为根。考虑到一个点$u$，如果在点$u$设置了守卫，那么从$u$的子节点$v$开始往下的$D-1$层（$v$为第$1$层开始）都会被保护，那么从$v$开始往下的$D-1$层，在某些条件下就不用设置守卫。所以建立一个DP模型：$f[u][i]$表示$u$的子树内，至少前$i$层已经被$u$的子树外的守卫保护，然后保护$u$的子树内所有应被保护的节点的最小代价。
但这样存在一个问题：如果$u$存在两个子节点$v$和$w$，并且$v$的子树内深度最小的守卫与$v$的距离为$k$且$k<D$，那么节点$u$，以及$w$的子树内前$D-k-1$层节点都会被保护，再放守卫有可能没有必要。所以再开一个数组：$g[u][i]$表示$u$的子树内，保护$u$的子树内所有应被保护的节点，并且往$u$的子树外至少保护了$i$层的最小代价，等价于$i=0$时，$u$的子树内深度最小的守卫与$u$的距离$\geq D$，否则深度最小的守卫与$u$的距离$\leq D-i$。
转移就是做一个树形背包DP。
一开始（只有$u$一个节点时）：
1、如果这个节点不需要被保护，那么$f[u][0]=g[u][0]=0$，否则$f[u][0]$和$g[u][0]$都为守卫点$u$的代价。
2、$\forall 1\leq i\leq D,f[u][i]=0$，$g[u][i]$为守卫点$u$的代价。
下面设当前位于点$u$，考虑到了$u$的子节点$v$，$f'[u][]$为点$u$已经遍历过的子树的$f$，$g'[u][]$同理。
一、$g$的转移：
（1）点$u$已经遍历过的子树向子树外的$i$层提供了保护。
此时转移为$g[u][i]=\min(g[u][i],g'[u][i]+f[v][i])$。
（2）点$v$的子树向子树外的$i+1$层提供了保护。
此时转移为$g[u][i]=\min(g[u][i],f'[u][i+1]+g[v][i+1])$。
（3）注意到“至少”二字：
$g[u][i]=\min(g[u][i],g[u][i+1]$。
二、$f$的转移：
1、$f[u][0]=g[u][0]$。
2、从$u$开始的$i$层被保护，那么从$v$开始的$i-1$层被保护，所以：
$f[u][i]=\min(f[u][i],f'[u][i]+f[v][i-1])$
3、同样，$f[u][i]=\min(f[u][i],f[u][i-1])$。
代码：

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
inline int read() {
    int res = 0; bool bo = 0; char c;
    while (((c = getchar()) < '0' || c > '9') && c != '-');
    if (c == '-') bo = 1; else res = c - 48;
    while ((c = getchar()) >= '0' && c <= '9')
        res = (res << 3) + (res << 1) + (c - 48);
    return bo ? ~res + 1 : res;
}
const int N = 5e5 + 5, M = 1e6 + 5, R = 23, INF = 0x3f3f3f3f;
int n, d, ecnt, nxt[M], adj[N], go[M], f[N][R], g[N][R], x[R], y[R], val[N];
bool vis[N];
void chkmin(int &a, int b) {if (b < a) a = b;}
void add_edge(int u, int v) {
    nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v;
    nxt[++ecnt] = adj[v]; adj[v] = ecnt; go[ecnt] = u;
}
void dfs(int u, int fu) {
    int i; f[u][0] = g[u][0] = vis[u] ? val[u] : 0;
    for (i = 1; i <= d; i++) f[u][i] = 0, g[u][i] = val[u];
    for (int e = adj[u], v; e; e = nxt[e]) {
        if ((v = go[e]) == fu) continue; dfs(v, u);
        for (i = 0; i <= d; i++) x[i] = y[i] = INF;
        for (i = d; i >= 0; i--) {
            chkmin(y[i], g[u][i] + f[v][i]);
            if (i < d) chkmin(y[i], g[v][i + 1] + f[u][i + 1]),
                chkmin(y[i], y[i + 1]);
        }
        x[0] = y[0]; for (i = 1; i <= d; i++) {
            chkmin(x[i], f[u][i] + f[v][i - 1]);
            chkmin(x[i], x[i - 1]);
        }
        for (i = 0; i <= d; i++) f[u][i] = x[i], g[u][i] = y[i];
    }
}
int main() {
    int i, u, v, tmp; n = read(); d = read();
    for (i = 1; i <= n; i++) val[i] = read(); tmp = read();
    for (i = 1; i <= tmp; i++) vis[read()] = 1;
    for (i = 1; i < n; i++) u = read(), v = read(), add_edge(u, v);
    printf("%d\n", (dfs(1, 0), f[1][0]));
    return 0;
}