dp,因为d很小,所以设
f
[
i
]
[
j
]
f[i][j]
f[i][j]为点
i
i
i,子树中深度为
j
j
j的点都不控制的代价
g
[
i
]
[
j
]
g[i][j]
g[i][j]为点
i
i
i,子树中深度为
j
j
j的点都被控制,且向上还有
j
j
j个被控制的代价
转移比较简单,注意
f
f
f和
g
g
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;
}