luogu 1040 加分二叉樹

題目描述 Description
設一個n個節點的二叉樹tree的中序遍歷爲(l,2,3,…,n),其中數字1,2,3,…,n爲節點編號。每個節點都有一個分數(均爲正整數),記第j個節點的分數爲di,tree及它的每個子樹都有一個加分,任一棵子樹subtree(也包含tree本身)的加分計算方法如下:

subtree的左子樹的加分× subtree的右子樹的加分+subtree的根的分數

若某個子樹爲主,規定其加分爲1,葉子的加分就是葉節點本身的分數。不考慮它的空

子樹。

試求一棵符合中序遍歷爲(1,2,3,…,n)且加分最高的二叉樹tree。要求輸出;

(1)tree的最高加分

(2)tree的前序遍歷

現在,請你幫助你的好朋友XZ設計一個程序,求得正確的答案。

輸入描述 Input Description
第1行:一個整數n(n<=30),爲節點個數。

第2行:n個用空格隔開的整數,爲每個節點的分數(分數<=100)

輸出描述 Output Description
第1行:一個整數,爲最高加分(結果不會超過4,000,000,000)。

第2行:n個用空格隔開的整數,爲該樹的前序遍歷。

樣例輸入 Sample Input
5

5 7 1 2 10

樣例輸出 Sample Output
145

3 1 2 4 5

數據範圍及提示 Data Size & Hint
n(n<=30)

分數<=100

這是個中序優先遍歷,我說不然哪來區間dp順序呢

然後就遞歸下去,執行區間dp,枚舉斷點,斷點就是根,根的兩邊就是左右子樹,然後…………這題就做完了……

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define int long long
using namespace std;
int f[40][40],n,anss,ge[40][40];
void print(int l,int r)
{
	if(l>r)return;
	if(l==r)
	{
		printf("%lld ",l);
		return;
	}
	printf("%lld ",ge[l][r]);
	print(l,ge[l][r]-1);
	print(ge[l][r]+1,r);
}
int dfs(int l,int r)
{
	if(f[l][r])return f[l][r];
	if(l==r)return f[l][l];
	if(l>r)return 1;
	for(int i=l;i<=r;i++)
	{
		int ans=dfs(l,i-1)*dfs(i+1,r)+f[i][i];
		if(ans>f[l][r])
		{
			f[l][r]=ans;
			ge[l][r]=i;
		}
	}
	return f[l][r];
}
signed main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&f[i][i]);
	}
	printf("%lld\n",dfs(1,n));
	print(1,n);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章