bzoj4557【JLOI2016】侦查守卫

这道题对于我来说并不是特别简单，还可以。

更新一下blog

树形DP

f[i][j]表示i的子树中，最高覆盖到i向下第j层的最小花费。

g[i][j]表示i的子树全部覆盖，还能向上覆盖j层的最小花费。

这样还是有许多细节的。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;

const int NN=500007,INF=1e9+7,DD=22;

int n,m,d;
int w[NN],g[NN][DD]={0},f[NN][DD]={0};
int cnt,head[NN],next[NN*2],rea[NN*2];
bool mark[NN];

void add(int u,int v)
{
    cnt++;
    next[cnt]=head[u];
    head[u]=cnt;
    rea[cnt]=v;
}
void dfs(int u,int fa)
{
    if (mark[u]) f[u][0]=g[u][0]=w[u];
    for (int i=1;i<=d;i++)
        g[u][i]=w[u];
    g[u][d+1]=INF;
    for (int i=head[u];i!=-1;i=next[i])
    {
        int v=rea[i];
        if (v==fa) continue;
        dfs(v,u);
        for (int j=d;j>=0;j--)
            g[u][j]=min(f[v][j]+g[u][j],f[u][j+1]+g[v][j+1]);
        for (int j=d;j>=0;j--)
            g[u][j]=min(g[u][j],g[u][j+1]);
        f[u][0]=g[u][0];
        for (int j=1;j<=d;j++)
            f[u][j]+=f[v][j-1];
        for (int j=1;j<=d;j++)
            f[u][j]=min(f[u][j-1],f[u][j]);            
    }    
}
int main()
{
    int x,y,z;
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&d);
    for (int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        scanf("%d",&x);
        mark[x]=true;
    }
    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]);
}