洛谷 P4436 [HNOI/AHOI2018]游戏(DP)

这是一道关于动态规划的编程题目,描述了一个基于树形结构的游戏问题。通过从根节点开始的深度优先搜索(DFS),定义f[i][j][k]表示到达第i个节点时,已修复j条公路和k条铁路的最小代价。转移方程涉及到两种情况的取最大值,复杂度为O(20000*40*40)。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

传送门


今天信心赛第一题,其实真的不是很难,只是我太弱不会。。
其实就是DP啦。
公路起点记为左儿子,铁路起点记为右儿子,从根开始深搜做DP,f[i][j][k]表示走到第i个节点,重修了j条公路与k条铁路的最小贡献。

转移方程:
f[i][j][k]=max(dfs(son[0],j+1,k)+dfs(son[1],j,k),dfs(son[0],j,k)+dfs(son[1],j,k+1)) f [ i ] [ j ] [ k ] = m a x ( d f s ( s o n [ 0 ] , j + 1 , k ) + d f s ( s o n [ 1 ] , j , k ) , d f s ( s o n [ 0 ] , j , k ) + d f s ( s o n [ 1 ] , j , k + 1 ) )

复杂度大概是 O(200004040) O ( 20000 ∗ 40 ∗ 40 )

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#define ll long long
using namespace std;

struct node{ll son[2];}tr[20010];
ll a[20010],b[20010],c[20010],f[20010][42][42];
ll n;

void read(ll &sum)
{
    char ch=getchar();
    sum=0;int f=0;
    while((ch<'0'||ch>'9')&&(ch!='-')) ch=getchar();
    f=((ch=='-')&&(ch=getchar()));
    while(ch>='0'&&ch<='9') sum=sum*10+(ch-48),ch=getchar();
    (f)&&(sum=-sum);
}

ll min(ll x,ll y)
{
    return x<y?x:y;
}

ll dfs(int i,int j,int k)
{
    if(i>=n) return c[i-n+1]*(a[i-n+1]+j)*(b[i-n+1]+k);
    if(f[i][j][k]<f[0][0][0]) return f[i][j][k]; 
    return f[i][j][k]=min(dfs(tr[i].son[0],j+1,k)+dfs(tr[i].son[1],j,k),dfs(tr[i].son[0],j,k)+dfs(tr[i].son[1],j,k+1));
}

int main()
{
    read(n);
    for(int i=1;i<n;i++)
    {
        ll x,y;
        read(x);read(y);
        if(x<0) x=-x-1+n;
        if(y<0) y=-y-1+n;
        tr[i].son[0]=x;
        tr[i].son[1]=y;
    }
    for(int i=1;i<=n;i++)
    {
        read(a[i]);read(b[i]);read(c[i]);
    }
    memset(f,63,sizeof(f));
    printf("%lld",dfs(1,0,0));
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值