力扣算法学习day14-3

106-从中序与后序遍历序列构造二叉树

题目

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        int[] index = new int[1];
        index[0] = postorder.length-1;
        return nodeTraversal(0,inorder.length-1,inorder,index,postorder);
    }
    public TreeNode nodeTraversal(int left,int right,int[] inorder,int[] index,int[] postorder){
        if(left > right){
            return null;
        }
        // 第一步，找到每个子树的根结点
        int rootValue = postorder[index[0]];
        TreeNode root = new TreeNode(rootValue);
        if(right == left){
            return root;
        }
        int rootIndex = 0;
        for(int i = left;i <= right;i++){
            if(inorder[i] == rootValue){
                rootIndex = i;
            }
        }

        // 第二步，求右左结点，如果没有长度了，就说明到底了，需要求的结点为空。
        // 注：这里需要先求右节点，因为后序遍历是左右中，反过来就是中右左，所以
        // 后序数组从后往前的话是往右走。
        index[0] = index[0] - 1;
        TreeNode rightNode = nodeTraversal(rootIndex+1,right,inorder,index,postorder);
        if(rightNode == null){
            index[0] = index[0] + 1;
        }
        index[0] = index[0] - 1;
        TreeNode leftNode = nodeTraversal(left,rootIndex-1,inorder,index,postorder);
        if(leftNode == null){
            index[0] = index[0] + 1;
        }

        // 将左节点连接到左边，右节点连接到右边。
        root.left = leftNode;
        root.right = rightNode;

        // 返回根结点
        return root;
    }
    
}

105-从前序与中序遍历序列构造二叉树

题目

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int[] index = new int[1];
        index[0] = 0;

        return nodeTraversal(0,preorder.length-1,inorder,index,preorder);
    }
    public TreeNode nodeTraversal(int left,int right,int[] inorder,int[] index,int[] preorder){
        // left > right说明结点为null，直接返回
        if(left > right){
            return null;
        }
        // 第一步，结点不为null，则求根结点
        int rootValue = preorder[index[0]];
        TreeNode root = new TreeNode(rootValue);
        // 如果 left = right 说明为叶子结点，直接返回。
        if(left == right){
            return root;
        }
        // 第二步，如果这个结点不为叶子结点，说明它是其他结点的父节点，需要找它在先序遍历数组中的坐标
        // 1,2  2,1
        int rootIndex = 0;
        for(int i = left;i <= right;i++){
            if(inorder[i] == rootValue){
                rootIndex = i;
            }
        }

        // 第三步，找到坐标后，需要遵循规律，因为已知的是前序和中序，所以，用前序作为按顺序出结点的数组，
        // 中序数组则用来定位，那么应该先向左再向右找结点。（即中序本身的顺序）
        index[0] = index[0] + 1;
        TreeNode leftNode = nodeTraversal(left,rootIndex-1,inorder,index,preorder);
        if(leftNode == null){
            index[0] = index[0] - 1;
        }

        index[0] = index[0] + 1;
        TreeNode rightNode = nodeTraversal(rootIndex+1,right,inorder,index,preorder);
        if(rightNode == null){
            index[0] = index[0] - 1;
        }

        // 第四步，连接左右结点
        root.left = leftNode;
        root.right = rightNode;

        return root;
    }
}