题意
给你
n−1
n
−
1
个城市,
n
n
个乡村是一个二叉树,一条边是公路,一条边是铁路,叶子是乡村,城市i向城市j连边必须满足 , 然后乡村有
3
3
个参数,要求每一个城市选一条通向该城市的路进行翻修(也就是翻修
n−1
n
−
1
条道路),设乡村到首都城市
1
1
的路上,未翻修的公路条数是,未翻修的铁路条数是
y
y
,每一个乡村的贡献是:
要求选择一个翻修方案使得所有乡村贡献的总和最小
分析
设
f[i][x][y]
f
[
i
]
[
x
]
[
y
]
表示在i城市内,上面还有x条公路,y条铁路,子树内的贡献
记忆化搜索就好了
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll inf = (ll) 1e18;
const ll N = 20010;
inline ll read()
{
ll p=0; ll f=1; char ch=getchar();
while(ch<'0' || ch>'9'){if(ch=='-') f=-1; ch=getchar();}
while(ch>='0' && ch<='9'){p=p*10+ch-'0'; ch=getchar();}
return p*f;
}
ll f[N][42][42];
ll a[N],b[N],c[N],n; ll son[N][2];
ll dfs(ll x,ll p,ll q)
{
if(x>=n) return c[x-n+1] * (p + a[x-n+1]) * (q + b[x-n+1]);
if(f[x][p][q] < inf) return f[x][p][q];
return f[x][p][q] = min( dfs(son[x][0],p+1,q) + dfs(son[x][1],p,q) , dfs(son[x][0],p,q) + dfs(son[x][1],p,q+1) ) ;
}
int main()
{
n = read(); memset(f,127,sizeof(f));
for(ll i=1;i<n;i++)
{
ll x = read(); ll y = read();
if(x<0) x=-x+(n-1); if(y<0) y=-y+(n-1);
son[i][0] = x; son[i][1] = y;
}for(ll i=1;i<=n;i++) a[i] = read() , b[i] = read() , c[i] = read();
return printf("%lld\n",dfs(1,0,0)),0;
}