[JLOI2016]侦察守卫树形DP

最新推荐文章于 2022-01-06 11:36:56 发布

原创最新推荐文章于 2022-01-06 11:36:56 发布 · 224 阅读

1 ·

CC 4.0 BY-SA版权

动态规划专栏收录该内容

22 篇文章

订阅专栏

博客围绕 JLOI2016 展开，提到自身解题能力不足。介绍了状态转移方程，如用 f[x][j] 表示从 x 开始向下 j 层（含 x）以下都被覆盖的最小花费，g[x][j] 表示 x 及子树覆盖完且 x 向上还能覆盖 j 层的最小花费，并给出了相应状态转移公式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[JLOI2016]

菜的真实，不会写
f[x][j]表示从x开始向下j层（包括x）以下都被覆盖的最小花费
g[x][j]表示x及x的子树都覆盖完了，x向上还能覆盖j层的最小花费
如果选x：g[x][j]=g[x][j]+f[y][j]
如果选y：g[x][j]=g[y][j+1]+f[x][j+1]
f[x][j]=\sum f[y][j-1]

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
using namespace std;
const int inf=1e9;
const int N=5e5+100;
struct node{int y,n;}e[N<<1];
int lin[N<<1],f[N][25],g[N][25],w[N],v[N],len,n,d,m,x,y;
inline int read(){
	int num=0;char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))num=num*10+ch-'0',ch=getchar();
	return num;
}
void LK(int x,int y)
{e[++len].y=y,e[len].n=lin[x],lin[x]=len;}
void init(){
	n=read(),d=read();
	rep(i,1,n)w[i]=read();
	m=read();
	rep(i,1,m)x=read(),v[x]=1;
	rep(i,2,n){
		x=read(),y=read();
		LK(x,y),LK(y,x);
	}
}
void dfs(int x,int fa){
	if(v[x])f[x][0]=g[x][0]=w[x];
	rep(i,1,d)g[x][i]=w[x]; g[x][d+1]=inf;
	for(int i=lin[x];i;i=e[i].n){
		int y=e[i].y;
		if(y==fa)continue;
		dfs(y,x);
		per(j,d,0)g[x][j]=min(g[x][j]+f[y][j],g[y][j+1]+f[x][j+1]);
		per(j,d,0)g[x][j]=min(g[x][j],g[x][j+1]);
		f[x][0]=g[x][0];
		rep(j,1,d+1)f[x][j]+=f[y][j-1];
		rep(j,1,d+1)f[x][j]=min(f[x][j],f[x][j-1]);
	}
}
int main()
{
	//freopen("a.in","r",stdin);
	init();
	dfs(1,1);
	cout<<f[1][0];
	return 0;
}