luogu P4438 [HNOI/AHOI2018]道路

题目传送门：https://www.luogu.org/problemnew/show/P4438

题意：

有n-1个点为城市，n-1个点为农村，每个城市连出两种道路，一边为公路，一边为铁路。现在每一个城市所连出的其中一种路都要修改，求通过公式（i表示第i个农村，x 表示到城市1（就是根）的未被修复的公路，y 表示到城市1（就是根）的未被修复的铁路）所算出的最小值。

思路：

树形dp。

考场竟然没有想出来，太菜。

son[i][0]表示i的左儿子（也就是修公路），son[i][1]表示i的右儿子（也就是修铁路）。

f[i][j][k]表示第i个点到根经过j个未被修复的公路，k个未被修复的铁路所得到的最小值。

最后用个记忆化搜索一下即可。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
	int n;
	struct node{LL x,y,z;} a[20010];
	int son[20010][5];
	LL f[20010][45][45];
LL dfs(int x,int p,int q)
{
	if(x>=n) return a[x-n+1].z*(a[x-n+1].x+p)*(a[x-n+1].y+q);
	if(f[x][p][q]!=f[n+1][41][41]) return f[x][p][q];
	return f[x][p][q]=min(dfs(son[x][0],p,q)+dfs(son[x][1],p,q+1),dfs(son[x][1],p,q)+dfs(son[x][0],p+1,q));
}
int main()
{
	int x,y;
	scanf("%d",&n);
	memset(f,63,sizeof(f));
	for(int i=1;i<n;i++)
	{
		scanf("%d %d",&x,&y);
		if(x<0) x=-x+n-1;
		if(y<0) y=-y+n-1;
		son[i][0]=x;
		son[i][1]=y;
	}
	for(int i=1;i<=n;i++)
		scanf("%lld %lld %lld",&a[i].x,&a[i].y,&a[i].z);
	printf("%lld",dfs(1,0,0));
}