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=,INF=1e9+,DD=;

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