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]);}