【题解】加分二叉树

加分二叉树

https://ac.nowcoder.com/acm/problem/16681

我横竖睡不着,仔细看了半夜,才从字缝里看出字来,满本都写着四个字是‘区间DP’!

为什么是区间DP呢?因为是中序遍历啊!中序遍历有一个很大的特点就是,根在中间,二叉树的左右子树节点在根的左右对应。

那么我们只需要枚举根就好了,枚举根是不是就很像区间DP。对于一个区间,我们枚举分割点就相当于枚举根,然后一步步转移就好了,同时也要记录下每个区间的根是什么,也就是最优值是多少。

既然是DP,那也写一个转移方程吧(感觉题目已经给了诶)
图片说明

因为dfs其实更好写,所以用dfs更方便。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstring>
#include<stack>
#define fs first
#define se second
#define pb push_back
#define cppio ios::sync_with_stdio(false);cin.tie(0)
#define eps 1e-7 
using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
typedef vector<int> VI;

const int maxn=1e2+6;
const ll inf=0x3f3f3f3f;
const ll mod=1e9+7;

ll a[maxn];
int n;
ll dp[maxn][maxn];
ll root[maxn][maxn];

ll dfs(int l,int r){
    if(dp[l][r]!=-1) return dp[l][r];
    if(l>r) return 1;
    if(l==r){
        root[l][r]=l;
        return a[l];
    } 
    dp[l][r]=0;
    for(int i=l;i<=r;i++){
        ll tmp=dfs(l,i-1)*dfs(i+1,r)+a[i];
        if(tmp>dp[l][r]){
            root[l][r]=i;
            dp[l][r]=tmp;
        }
    }
    return dp[l][r];
}

void dfs2(int l,int r){
    if(l>r) return ;
    printf("%lld ",root[l][r]);
    dfs2(l,root[l][r]-1);
    dfs2(root[l][r]+1,r);
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%lld",a+i);
    }
    memset(dp,-1,sizeof(dp));
    printf("%lld\n",dfs(1,n));
    dfs2(1,n);
    return 0;
}
全部评论

相关推荐

头像
昨天 14:28
长沙理工大学
刷算法真的是提升代码能力最快的方法吗?&nbsp;刷算法真的是提升代码能力最快的方法吗?
牛牛不会牛泪:看你想提升什么,代码能力太宽泛了,是想提升算法能力还是工程能力? 工程能力做项目找实习,算法也分数据结构算法题和深度学习之类算法
点赞 评论 收藏
分享
废铁汽车人:秋招真是牛鬼蛇神齐聚一堂
点赞 评论 收藏
分享
11-15 17:19
湖南大学 Java
成果成果成果果:这是哪个公司的hr,这么离谱吗,我没见过用性别卡技术岗的,身边女性同学拿大厂offer的比比皆是
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务