bzoj 4557: [JLoi2016]侦察守卫

题意：

一棵树中有些黑点，在每个点上放守卫要花一定代价，守卫侦查范围为d，求覆盖所有黑点最小代价。

题解：

$f[x][i]$表示覆盖x的子树中与x的距离超过i的黑点的代价。（可以有小于i的点也被覆盖）
$g[x][i]$表示完整x的子树后，还能向上至少i步的最小代价。
转移留坑，先上代码。
code：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
using namespace std;
const int inf=1000000000;
int n,d,w[500010];
bool mark[500010];
struct node{
    int y,next;
}a[1000010];int len=0,last[500010];
int f[500010][25],g[500010][25];//f[x][i]x走i步走不到的点覆盖。 
void ins(int x,int y)
{
    a[++len].y=y;
    a[len].next=last[x];last[x]=len;
}
void dfs(int x,int fa)
{
    if(mark[x]) f[x][0]=g[x][0]=w[x];
    for(int i=1;i<=d;i++) g[x][i]=w[x];
    g[x][d+1]=inf;
    for(int i=last[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(y==fa) continue;
        dfs(y,x);
        for(int j=d;j>=0;j--) g[x][j]=min(g[x][j]+f[y][j],g[y][j+1]+f[x][j+1]);
        for(int j=d;j>=0;j--) g[x][j]=min(g[x][j],g[x][j+1]);
        f[x][0]=g[x][0];
        for(int j=1;j<=d+1;j++) f[x][j]+=f[y][j-1];
        for(int j=1;j<=d+1;j++) f[x][j]=min(f[x][j],f[x][j-1]);
    }
}
int main()
{
    scanf("%d %d",&n,&d);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    memset(mark,false,sizeof(mark));
    int m;scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int x;scanf("%d",&x);
        mark[x]=true;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;scanf("%d %d",&x,&y);
        ins(x,y);ins(y,x);
    }
    dfs(1,0);
    printf("%d",f[1][0]);
}