求和树

今日开始 每日至少一题

原题链接

题目描述
给满出二叉树，编写算法将其转化为求和树

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

二叉树：
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

输出
复制

0 4 0 20 0 12 0

import java.util.*;
 

public class Main {
    public static void main(String[] args) {

        Scanner sc  = new Scanner(System.in);
        sc.nextLine();
        String[] strNodes = sc.nextLine().split(" ");
        //中序遍历的节点
         
        int[] nodes = Arrays.stream(strNodes)
                .mapToInt(Integer::parseInt)
                .toArray();
         
        //二分法
        handle(nodes,0,nodes.length-1);



        for(int i:nodes) {
            System.out.print(i+" ");
        }




    }



    //处理一颗小树 ---> 处理整个树
    static int handle(int[]nodes, int l,int r) {

        if(l==r) {
            int temp = nodes[l];
            nodes[l] = 0;
            return temp;
        }
        int mid = (l+r)/2;
        int left = handle(nodes,l,mid-1);
        int right = handle(nodes,mid+1,r);
        int midVal = nodes[mid];
        nodes[mid] = left+right;
        return midVal+nodes[mid];
    }


 












}

思路总结：

1. 满二叉树，直接中序遍历即可确定整棵树
2. 递归法，类似于给一棵二叉树求和，不过改了一下形式