[BZOJ4557][树形DP]JLOI2016：侦察守卫

版权声明：虽然博主很菜，但是还是请注明出处（我觉得应该没人偷我的博客） https://blog.csdn.net/qq_43346903/article/details/88982031

BZOJ4557

dp，因为d很小，所以设 $f[i][j]$为点 $i$，子树中深度为 $j$的点都不控制的代价
$g[i][j]$为点 $i$，子树中深度为 $j$的点都被控制，且向上还有 $j$个被控制的代价
转移比较简单，注意 $f$和 $g$可以从自己转移过来

Code：

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=5e5+5,L=25,INF=1e9;
int f[N][L],g[N][L];
int vis[N<<1],nxt[N<<1],head[N],tot=0;
inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
int pt[N];
int n,d,need[N],w[N];
void dfs(int v){
	if(need[v]) f[v][0]=g[v][0]=w[v];
	for(int i=1;i<=d;i++) g[v][i]=w[v];
	g[v][d+1]=INF;
	pt[v]=1;
	for(int i=head[v];i;i=nxt[i]){
		int y=vis[i];
		if(pt[y]) continue;
		dfs(vis[i]);
		for(int j=d;j>=0;j--) g[v][j]=min(g[v][j]+f[y][j],g[y][j+1]+f[v][j+1]);
		for(int j=d;j>=0;j--) g[v][j]=min(g[v][j],g[v][j+1]);
		f[v][0]=g[v][0];
		for(int j=1;j<=d+1;j++) f[v][j]+=f[y][j-1];
		for(int j=1;j<=d+1;j++) f[v][j]=min(f[v][j],f[v][j-1]);
	}
}
int main(){
	n=read();d=read();
	for(int i=1;i<=n;i++) w[i]=read();
	int m=read();
	for(int i=1;i<=m;i++) need[read()]=1;
	for(int x,y,i=1;i<n;i++){x=read();y=read();add(x,y);add(y,x);}
	dfs(1);
	cout<<f[1][0];
	return 0;
}