Acwing 479.加分二叉樹(區間dp)

在這裏插入圖片描述
當看到這個的時候,我是不知道怎麼遍歷這個二叉樹(儘管給我了中序遍歷)
後來我才知道一箇中序遍歷是無法確定二叉樹的
老規矩
老師的視頻網址:



https://www.acwing.com/video/495/

老師用了區間dp;

dp[l][r]是左邊界l,右邊界r的最大加分
同時還需要設一個k表示根節點,然後dp[l][k-1]表示左子樹最大加分,dp[k+1][r]表示有指數最大加分。然後遍歷這
一區間最大的加分就可,題目中要求輸出一個前序的序列,那我們把每一個最大區間也就是dp[l][k-1]*dp[k+1][r]+w[k]
的根節點記下來就好了用g[l][r]來保存.

最後說一下題意

對與一個根節點
1.若只有一個子樹,則根的加分爲子樹的分值*1+根的分數。
2.若是有兩個子樹,則根的加分爲左子樹分值*右子樹分值+根的分數
3.若沒有子樹,則根的加分爲根的分數本身。

最後說下代碼吧,也不是很長

#include<iostream>
using namespace std;
const int N=35;
int w[N],dp[N][N],g[N][N];
void print(int l,int r)//輸出前序 根、左、右
{
   
   
    if(l>r)
    return;
    cout<<g[l][r]<<" ";
    print(l,g[l][r]-1);
    print(g[l][r]+1,r);
    return;
}
int main(void)
{
   
   
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>w[i];
    for(int len=1;len<=n;len++)
    for(int l=1;l+len-1<=n;l++)
    {
   
   
       int  r=l+len-1;
        if(len==1)
        {
   
   
            dp[l][r]=w[l];
            g[l][r]=l;
        }
        else
        {
   
   
            for(int k=l;k<=r;k++)
            {
   
   
                int left=k==l?1:dp[l][k-1];
                int right=k==r?1:dp[k+1][r];
                int score=left*right+w[k];
                if(score>dp[l][r])
                {
   
   
                    dp[l][r]=score;
                    g[l][r]=k;
                }
            }
        }
    }
    cout<<dp[1][n]<<endl;
    print(1,n);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章