當看到這個的時候,我是不知道怎麼遍歷這個二叉樹(儘管給我了中序遍歷)
後來我才知道一箇中序遍歷是無法確定二叉樹的
老規矩
老師的視頻網址:
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);
}