bzoj4557 [JLoi2016]侦察守卫——DP

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4557

见这位的博客：https://www.cnblogs.com/Narh/p/9140326.html

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
int const maxn=,inf=1e9;
ll w[maxn],f[maxn][],g[maxn][];
int n,d,m,head[maxn],ct;//f向下 g向上
bool mark[maxn];
struct N{
    int to,next;
    N(int t=,int n=):to(t),next(n) {}
}edge[maxn<<];
void dp(int x,int fa)
{
    if(mark[x])f[x][]=w[x],g[x][]=w[x];
    else f[x][]=,g[x][]=;
    for(int i=;i<=d;i++)g[x][i]=w[x];
    g[x][d+]=inf;
    for(int i=head[x];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa)continue;
        dp(v,x);
        for(int i=;i<=d;i++)g[x][i]=min(g[x][i]+f[v][i],g[v][i+]+f[x][i+]);
        for(int i=d;i;i--)g[x][i-]=min(g[x][i],g[x][i-]);//倒序需要
        f[x][]=g[x][];
        for(int i=;i<=d;i++)f[x][i]+=f[v][i-];
        for(int i=;i<=d;i++)f[x][i]=min(f[x][i],f[x][i-]);
    }
}
int main()
{
    scanf("%d%d",&n,&d);
    for(int i=;i<=n;i++)scanf("%lld",&w[i]);
    scanf("%d",&m);
    for(int i=,x;i<=m;i++)
    {
        scanf("%d",&x);
        mark[x]=;
    }
    for(int i=,x,y;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        edge[++ct]=N(y,head[x]);head[x]=ct;
        edge[++ct]=N(x,head[y]);head[y]=ct;
    }
    dp(,);
    printf("%lld",f[][]);
    return ;
}