[HNOI/AHOI2018]道路

题目

https://www.luogu.org/problemnew/show/P4438

思路

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

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

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

代码

#include  
#include  
#include  
#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