刷题记录-重建二叉树

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

解题思路:前序遍历的第一个值是根节点,在中序遍历中根节点左边是左子树,右边是有字数,然后递归

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode root = reConstructBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
        return root;
    }
    private TreeNode reConstructBinaryTree(int[] pre, int startPre, int endPre, int[] in, int startin, int endin){
        if(startPre>endPre||startin>endin)
            return null;
        TreeNode root = new TreeNode(pre[startPre]);
        
        for(int i = startin;i <= endin; i++){
            if(in[i] == pre[startPre]){
                root.left = reConstructBinaryTree(pre, startPre+1, startPre+i-startin, in, startin, i-1);
                root.right = reConstructBinaryTree(pre, startPre+i-startin+1, endPre, in, i+1, endin);
            
                break;
            }
        }
        return root;
    }
}

总结:二叉树的题都离不开框架

void traverse(TreeNode root) {
    // 前序遍历
    traverse(root.left)
    // 中序遍历
    traverse(root.right)
    // 后序遍历
}

猜你喜欢

转载自www.cnblogs.com/tendermelon/p/12763633.html
今日推荐