bzoj 5290: [Hnoi2018]道路 记忆化搜索

题意

给你$n-1$个城市，$n$个乡村是一个二叉树，一条边是公路，一条边是铁路，叶子是乡村，城市i向城市j连边必须满足$i > j$ , 然后乡村有$3$个参数$a[i],b[i],c[i]$，要求每一个城市选一条通向该城市的路进行翻修（也就是翻修$n-1$条道路），设乡村到首都城市$1$的路上，未翻修的公路条数是$x$，未翻修的铁路条数是$y$，每一个乡村的贡献是：$c[i](a[i]+x)(b[i]+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;
}