[bzoj4557]侦察守卫

令g[i][j]表示覆盖了i的子树中距离i大于等于j的所有点，f[i][j]表示覆盖了i的子树和子树外距离i小于等于j的所有点，有递推式$f[i][j]=min(f[i][j]+g[son][j],f[son][j+1]+g[i][j+1]),g[i][j]+=g[son][j-1]$，特别的有g[i][0]=f[i][0]

#include<bits/stdc++.h>
using namespace std;
#define N 500005
struct ji{
    int nex,to;
}edge[N<<1];
int E,n,d,x,y,w[N],vis[N],head[N],f[N][31],g[N][31];
void add(int x,int y){
    edge[E].nex=head[x];
    edge[E].to=y;
    head[x]=E++;
}
void dfs(int k,int fa){
    if (vis[k])f[k][0]=g[k][0]=w[k];
    for(int i=1;i<=d;i++)f[k][i]=w[k];
    f[k][d+1]=0x3f3f3f3f;
    for(int i=head[k];i!=-1;i=edge[i].nex)
        if (edge[i].to!=fa){
            int v=edge[i].to;
            dfs(v,k);
            for(int j=0;j<=d;j++)f[k][j]=min(f[k][j]+g[v][j],f[v][j+1]+g[k][j+1]);
            for(int j=d;j>=0;j--)f[k][j]=min(f[k][j],f[k][j+1]);
            g[k][0]=f[k][0];
            for(int j=1;j<=d;j++)g[k][j]+=g[v][j-1];
            for(int j=1;j<=d;j++)g[k][j]=min(g[k][j],g[k][j-1]);
        }
}
int main(){
    scanf("%d%d",&n,&d);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    scanf("%d",&x);
    for(int i=1;i<=x;i++){
        scanf("%d",&y);
        vis[y]=1;
    }
    memset(head,-1,sizeof(head));
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    printf("%d",f[1][0]);
}