将满二叉树转换为求和树

给满出二叉树,编写算法将其转化为求和树

什么是求和树:二叉树的求和树, 是一颗同样结构的二叉树,其树中的每个节点将包含原始树中的左子树和右子树的和。

二叉树:
                  10
               /      \
             -2        6
           /   \      /  \ 
          8    -4    7    5

求和树:
                 20(4-2+12+6)
               /      \
           4(8-4)      12(7+5)
            /   \      /  \ 
          0      0    0    0

二叉树给出前序和中序输入,求和树要求中序输出;
所有处理数据不会大于int;


输入描述:
2行整数,第1行表示二叉树的前序遍历,第2行表示二叉树的中序遍历,以空格分割

输出描述:
1行整数,表示求和树的中序遍历,以空格分割

输入例子1:
10 -2 8 -4 6 7 5
8 -2 -4 10 7 6 5

输出例子1:
0 4 0 20 0 12 0

import java.util.Scanner;

public class Main {
	
	public static class TreeNode{
		int value;
		TreeNode left;
		TreeNode right;
		int old_value;
 		TreeNode(int value, TreeNode left, TreeNode right){
			this.value = value;
			this.left = left;
			this.right = right;
		}
	}
	
	public static TreeNode construct(int[] pre, int p_start, int p_end, int[] in, int i_start, int i_end) {
		if(p_end<p_start || i_start>i_end) {
			return null;
		}
		int index = -1;
		TreeNode root = new TreeNode(pre[p_start], null, null);
		for(int i=i_start; i<=i_end; i++) {
			if(in[i]==pre[p_start])
				index = i;
		}
		root.left = construct(pre, p_start+1, p_start+index-i_start, in, i_start, index-1);
		root.right = construct(pre, p_start+index-i_start+1, p_end, in, index+1, i_end);
		return root;
		
	}
	
	public static void back_order(TreeNode root) {
		if(root==null)
			return;
		if(root.left==null && root.right==null)
				{root.old_value = root.value; root.value=0; return;}
		back_order(root.left);
		back_order(root.right);
		int old_left_value = (root.left!=null)? root.left.old_value:0;
		int old_right_value = (root.right!=null)? root.right.old_value: 0;
		root.old_value = root.value + old_left_value + old_right_value;
		root.value = old_left_value + old_right_value;
		
	}
	
	public static void in_print(TreeNode root) {
		if(root==null)
			return;
		in_print(root.left);
		System.out.print(root.value);
		System.out.print(" ");
		in_print(root.right);
	}
	
	public static void main(String[] args) {
		
		Scanner scan = new Scanner(System.in);
		String pre_str = scan.nextLine();
		String[] p = pre_str.split(" ");
		int [] pre = new int[p.length];
		int i = 0;
		for(String s: p) {
			pre[i++] = Integer.parseInt(s);
		}
		String in_str = scan.nextLine();
		String [] in = in_str.split(" ");
		i = 0;
		int [] in_list = new int[in.length];
		for(String s: in) {
			in_list[i++] = Integer.parseInt(s);
		}
		
		TreeNode root = construct(pre, 0, pre.length-1, in_list, 0, in_list.length-1);
		back_order(root);
		in_print(root);
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值