[HNOI/AHOI2018]道路

Description:

W 国的交通呈一棵树的形状。W 国一共有(n - 1)个城市和(n)个乡村，其中城市从(1)到(n - 1) 编号，乡村从(1)到(n)编号，且(1)号城市是首都。道路都是单向的，本题中我们只考虑从乡村通往首都的道路网络。对于每一个城市，恰有一条公路和一条铁路通向这座城市。对于城市i， 通向该城市的道路（公路或铁路）的起点，要么是一个乡村，要么是一个编号比(i)大的城市。 没有道路通向任何乡村。除了首都以外，从任何城市或乡村出发只有一条道路；首都没有往 外的道路。从任何乡村出发，沿着唯一往外的道路走，总可以到达首都。
W 国的国王小 W 获得了一笔资金，他决定用这笔资金来改善交通。由于资金有限，小 W 只能翻修(n - 1)条道路。小 W 决定对每个城市翻修恰好一条通向它的道路，即从公路和铁 路中选择一条并进行翻修。小 W 希望从乡村通向城市可以尽可能地便利，于是根据人口调 查的数据，小 W 对每个乡村制定了三个参数，编号为(i)的乡村的三个参数是(a_i)，(b_i)和(c_i)。假设 从编号为(i)的乡村走到首都一共需要经过(x)条未翻修的公路与(y)条未翻修的铁路，那么该乡村 的不便利值为

[c_i cdot (a_i + x) cdot (b_i + y) ]

在给定的翻修方案下，每个乡村的不便利值相加的和为该翻修方案的不便利值。 翻修(n - 1)条道路有很多方案，其中不便利值最小的方案称为最优翻修方案，小 W 自然 希望找到最优翻修方案，请你帮助他求出这个最优翻修方案的不便利值。

Hint:

对于所有的数据，(n le 20000)，(1 le a_i,b_i le 60)，(1 le c_i le 10^9)，(s_i,t_i)是([-n,-1] cup (i,n - 1])内的整数，任意乡村可以通过不超过40条道路到达首都。

Solution:

表面上不可做实际上并不难的树型dp
设 (f[i][j][k]) 表示在i节点,到根修了j条公路,k条铁路的代价
显然有转移:
(f[i][j][k]=min(f[ls][j+1][k]+f[rs][j][k],f[ls][j][k],f[rs][j][k+1]))
可以考虑直接出叶子节点的f值,然后由根的初始状态出发(f[1][0][0]),记忆化搜索
但是这题卡空间,但是树的最大深度为40,我们发现一个点只由儿子转移过来,所以只需开一条链的空间大小,详见代码:

#include <map>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define ls p<<1 
#define rs p<<1|1
using namespace std;
typedef long long ll;
const int mxn=4e4+5;
int n,m,a[mxn],b[mxn],c[mxn],to[mxn][2];
ll f[105][85][85];

inline int read() {
    char c=getchar(); int x=0,f=1;
    while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
    while(c<='9'&&c>='0') {x=(x<<3)+(x<<1)+(c&15);c=getchar();}
    return x*f;
}
inline int chkmax(int &x,int y) {if(x<y) x=y;}
inline int chkmin(int &x,int y) {if(x>y) x=y;}

struct ed {
    int to,nxt;
}t[mxn<<1];

void dfs(int u,int id,int x,int y)
{
    if(to[u][0]) dfs(to[u][0],id+1,x+1,y);
    if(to[u][1]) dfs(to[u][1],id+2,x,y+1);
    if(u>n) {
        for(int i=0;i<=x;++i)
            for(int j=0;j<=y;++j)
                f[id][i][j]=1ll*c[u]*(a[u]+i)*(b[u]+j);
    }
    else {
        for(int i=0;i<=x;++i) 
            for(int j=0;j<=y;++j)
                f[id][i][j]=min(f[id+1][i+1][j]+f[id+2][i][j],f[id+1][i][j]+f[id+2][i][j+1]);
    }
}

int main()
{
    int u,v;
    n=read(); memset(f,0x3f,sizeof(f));
    for(int i=1;i<n;++i) {
        u=read(); v=read();
        if(u<0) u=-u+n;
        if(v<0) v=-v+n;
        to[i][0]=u; to[i][1]=v;
    }
    for(int i=1;i<=n;++i) 
        a[i+n]=read(),b[i+n]=read(),c[i+n]=read();
    dfs(1,1,0,0);
    printf("%lld",f[1][0][0]);
    return 0;
}