[JLOI2016]侦察守卫 树形DP

Description
给你一棵树，你选择一个点进行覆盖那么他就会覆盖所有距离他为D的节点，覆盖每个点有代价，有一些点必须被覆盖，要求最小代价。

---

Sample Input
12 2
8 9 12 6 1 1 5 1 4 8 10 6
10
1 2 3 5 6 7 8 9 10 11
1 3
2 3
3 4
4 5
4 6
4 7
7 8
8 9
9 10
10 11
11 12

---

Sample Output
10

---

我是来学习的。。。
设f[i][j]为i以下的子树全部覆盖，还往上覆盖j层的最小代价。
设g[i][j]为i一下有j层没有被覆盖的最小代价。
转移为：
f[x][j]=min(f[x][j]+g[x][j],g[x][j+1]+f[y][j+1])
g则直接从j-1转移即可。

---

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
typedef long long LL;
int _min(int x, int y) {return x < y ? x : y;}
int _max(int x, int y) {return x > y ? x : y;}
int read() {
	int s = 0, f = 1; char ch = getchar();
	while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
	while(ch >= '0' && ch <= '9') s = s * 10 + ch - '0', ch = getchar();
	return s * f;
}

struct edge {
	int x, y, next;
} e[1000010]; int len, last[500010];
int K, a[510000], f[21][500010], g[21][500010];
bool v[500010];
//f表示往上拓展，g往下缺

void ins(int x, int y) {
	e[++len].x = x, e[len].y = y;
	e[len].next = last[x], last[x] = len;
}

void treedp(int x, int fa) {
	if(v[x]) f[0][x] = g[0][x] = a[x];
	for(int i = 1; i <= K; i++) f[i][x] = a[x];
	f[K + 1][x] = 999999999;
	for(int k = last[x]; k; k = e[k].next) {
		int y = e[k].y;
		if(y != fa) {
			treedp(y, x);
			for(int i = K; i >= 0; i--) f[i][x] = _min(f[i][x] + g[i][y], g[i + 1][x] + f[i + 1][y]);
			for(int i = K; i >= 0; i--) f[i][x] = _min(f[i][x], f[i + 1][x]);
			g[0][x] = f[0][x];
			for(int i = 1; i <= K; i++) g[i][x] += g[i - 1][y];
			for(int i = 1; i <= K; i++) g[i][x] = _min(g[i][x], g[i - 1][x]);
		}
	}
}

int main() {
	int n = read(); K = read();
	for(int i = 1; i <= n; i++) a[i] = read();
	int m = read();
	for(int i = 1; i <= m; i++) v[read()] = 1;
	for(int i = 1; i < n; i++) {
		int x = read(), y = read();
		ins(x, y), ins(y, x);
	} treedp(1, 0);
	printf("%d\n", f[0][1]);
	return 0;
}