[bzoj4557][树形DP]侦察守卫

Description

小R和B神正在玩一款游戏。这款游戏的地图由N个点和N-1条无向边组成，每条无向边连接两个点，且地图是连通的
。换句话说，游戏的地图是一棵有N个节点的树。游戏中有一种道具叫做侦查守卫，当一名玩家在一个点上放置侦
查守卫后，它可以监视这个点以及与这个点的距离在D以内的所有点。这里两个点之间的距离定义为它们在树上的
距离，也就是两个点之间唯一的简单路径上所经过边的条数。在一个点上放置侦查守卫需要付出一定的代价，在不
同点放置守卫的代价可能不同。现在小R知道了所有B神可能会出现的位置，请你计算监视所有这些位置的最小代价 。

Input

第一行包含两个正整数N和D，分别表示地图上的点数和侦查守卫的视野范围。约定地图上的点用1到N的整数编号。
第二行N个正整数，第i个正整数表示在编号为i的点放置侦查守卫的代价Wi。保证Wi≤1000。第三行一个正整数M，
表示B神可能出现的点的数量。保证M≤N。第四行M个正整数，分别表示每个B神可能出现的点的编号，从小到大不
重复地给出。接下来N–1行，每行包含两个正整数U,V，表示在编号为U的点和编号为V的点之间有一条无向边。N<= 500000,D<=20

Output

仅一行一个整数，表示监视所有B神可能出现的点所需要的最小代价

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

题解

这题超强
我只想到了设g[i][j]为以第i个节点为根的子树，除了底下j层其他全部安全
然后就不会转移了。。
%一发发现，原来还要多一个f[i][j]表示以第i个节点为子树 子树安全，还能向上看j层
这就可以转移了嘛。
设遍历到x，当前访问儿子y
f[x][i]=min(g[x][i+1]+f[y][i+1],f[x][i]+g[y][i-1])
第一个表示说x原本不安全，在y的看管下变得安全了，且由于y的看管，还能往上再看i层
第二个表示说x原本就安全了，还能再把y看管的安全起来
很容易可以发现g[x][i]随着i的增大是单调递减的，f[x][i]反之，所以第一个转移显而易见正确
g[x][i]+=g[y][i-1]
这是维护g数组，因为你要保证下面都是安全的，所以只能g数组加g数组啦

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct node
{
    int x,y,next;
}a[1110000];int len,last[510000];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int f[510000][25],g[510000][25],w[510000];
int n,D;
bool v[510000];
void treedp(int x,int fa)
{
    if(v[x])f[x][0]=g[x][0]=w[x];
    for(int i=1;i<=D;i++)f[x][i]=w[x];
    f[x][D+1]=999999999;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa)
        {
            treedp(y,x);
            for(int i=D;i>=0;i--)f[x][i]=min(f[x][i]+g[y][i],g[x][i+1]+f[y][i+1]);
            for(int i=D;i>=0;i--)f[x][i]=min(f[x][i],f[x][i+1]);
            g[x][0]=f[x][0];
            for(int i=1;i<=D;i++)g[x][i]+=g[y][i-1];
            for(int i=1;i<=D;i++)g[x][i]=min(g[x][i],g[x][i-1]);
        }
    }
}
int main()
{
    scanf("%d%d",&n,&D);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    memset(v,false,sizeof(v));
    int m,op;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){scanf("%d",&op);v[op]=true;}
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    treedp(1,0);
    printf("%d\n",f[1][0]);
    return 0;
}