DP-加分二叉树(区间动态规划)

P1040 加分二叉树 https://www.luogu.org/problemnew/show/1040

题目描述

设一个n个节点的二叉树tree的中序遍历为(1,2,3,…,n),其中数字1,2,3,…,n为节点编号。每个节点都有一个分数(均为正整数),记第i个节点的分数为di,tree及它的每个子树都有一个加分,任一棵子树subtree(也包含tree本身)的加分计算方法如下:

subtree的左子树的加分× subtree的右子树的加分+subtree的根的分数。

若某个子树为空,规定其加分为1,叶子的加分就是叶节点本身的分数。不考虑它的空子树。

试求一棵符合中序遍历为(1,2,3,…,n)且加分最高的二叉树tree。要求输出;

(1)tree的最高加分

(2)tree的前序遍历

输入输出格式

输入格式:

第1行:一个整数n(n<30),为节点个数。

第2行:n个用空格隔开的整数,为每个节点的分数(分数<100)。

输出格式:

第1行:一个整数,为最高加分(结果不会超过4,000,000,000)。

第2行:n个用空格隔开的整数,为该树的前序遍历。

分析:

1.结合问题,如果整棵树的权值最大,必然有左子树的权值最大,右子树的权值也最大,符合最优性原理。

可以用区间动规模型解决。可得到状态转移方程:f[i][j]=max((f[i][k-1]*f[i+1][j]+f[k][k]),f[i][j])由此可以得出最高加分。

2.输出该树的前序遍历的序列,首先是要构建这个树。题目中所给的是二叉树中序遍历的序列,根据二叉树的特性建立树,只要边动规

边记录root[i][j]=k,k即i,j之间的根节点。最后根据前序遍历的规则输出相应序列即可。

import java.util.*;
public class 加分二叉树 {
	static int n;
	static int[][] f=new int[35][35];    //记录每个节点的分数
	static int[][] root=new int[35][35];//root[a][b],表示a,b之间的根节点编号
	public static int get_max(int l,int r) {
		//如果是空树
		if(l>r)
			return 1;
		//如果树不空
		if(f[l][r]==0) {
			//遍历左右节和根节点,得出加分值
			for(int i=l;i<=r;i++) {
				int t=f[l][r];
				//得到最大加分值
				f[l][r]=Math.max((get_max(l,i-1)*get_max(i+1,r)+f[i][i]),f[l][r]);
				if(t<f[l][r])
					root[l][r]=i;
			}
		}
		return f[l][r];
	}
	public static void dfs(int l,int r) {
		if(l>r)
			return;
		else {
			System.out.print(root[l][r]+" ");
			dfs(l,root[l][r]-1);
			dfs(root[l][r]+1,r);
		}
	}
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		n=in.nextInt();
		for(int i=1;i<=n;i++) {
			f[i][i]=in.nextInt();
			root[i][i]=i;
		}
		System.out.println(get_max(1,n));
		dfs(1,n);
	}
}



猜你喜欢

转载自blog.csdn.net/zw159357/article/details/79653358