[BZOJ5290][HNOI2018]道路(DP)

传送门


Day2的题目顺序很有问题啊,竟然把最简单的题放T3。。
不难发现是一个完全二叉树,然后公路连左孩子,铁路连右孩子。
就是一个类似于一个树形dp的东西,突破的关键就在原题中的一个隐晦的条件

任意乡村可以通过不超过40条道路到达首都。

那么我们就可以dp:到每个乡村,需要经过多少条公路多少条铁路空间复杂度是20000*40*40的样子,时间复杂度也是相同的。转移用记忆化搜索比较方便。


#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
inline ll read()
{
    ll x=0,f=1; char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
const int N=20010;
const ll INF=(ll)1e18;
ll f[N][41][41];
ll son[N][2],a[N],b[N],c[N];
int n;
ll dfs(int now,int x,int y)//当前now经过了x条公路,y条铁路
{
    if(now>=n) return c[now-n+1]*(a[now-n+1]+x)*(b[now-n+1]+y);
    if(f[now][x][y]<INF) return f[now][x][y];
    return f[now][x][y]=min(dfs(son[now][0],x+1,y)+dfs(son[now][1],x,y),dfs(son[now][0],x,y)+dfs(son[now][1],x,y+1));
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        ll u=read(),v=read();
        if(u<0) u=-u+(n-1); 
        if(v<0) v=-v+(n-1);
        son[i][0]=u,son[i][1]=v;
    }
    for(int i=1;i<=n;i++) a[i]=read(),b[i]=read(),c[i]=read();
    memset(f,127,sizeof(f));
    printf("%lld\n",dfs(1,0,0));
    return 0;
}

猜你喜欢

转载自blog.csdn.net/cabi_zgx/article/details/80053058