bzoj5290: [Hnoi2018]道路【树形dp/记忆化搜索】

传送门

解题思路:

考试时刚看以为是神仙题,于是开始打dfs暴力,然后猛然发现可以记忆化,然后就没有然后了……这不是普及组dp吗?
f [ i ] [ x ] [ y ] 表示以 i 为根的子树,上面有 x 条公路, y 条铁路未翻新的最优值,记忆化搜索即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=40005;
int n,l[N],r[N],a[N],b[N],c[N];
ll f[N/2][41][41];
ll dfs(int u,int x,int y)
{
    if(u>=n)return 1ll*(a[u]+x)*(b[u]+y)*c[u];
    if(f[u][x][y])return f[u][x][y];
    return f[u][x][y]=min(dfs(l[u],x+1,y)+dfs(r[u],x,y),dfs(l[u],x,y)+dfs(r[u],x,y+1));
}
int main()
{
    //freopen("road.in","r",stdin);
    //freopen("road.out","w",stdout);
    n=getint();
    for(int i=1,x;i<n;i++)
    {
        x=getint();
        l[i]=x>0?x:n-1-x;
        x=getint();
        r[i]=x>0?x:n-1-x;
    }
    for(int i=n;i<2*n;i++)a[i]=getint(),b[i]=getint(),c[i]=getint();
    cout<<dfs(1,0,0)<<'\n';
    return 0;
}

猜你喜欢

转载自blog.csdn.net/cdsszjj/article/details/80243280