二叉树练习（2）

Pennywise.

于 2023-02-25 23:02:06 发布

阅读量66

点赞数

分类专栏： Java练习题文章标签：数据结构算法 java

本文链接：https://blog.csdn.net/qq_52134698/article/details/129218435

版权

Java练习题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

一二叉树的层序遍历

二从前序和中序遍历中构造二叉树

三中序和后序遍历构造二叉树

一二叉树的层序遍历

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。（即逐层地，从左到右访问所有节点）

思路：

这里我们创建一个list0 = new List<List<Integer>>,和一个list = new List<Integer>,再创建一个que = new Queue<Integer>;先将root放入que中，然后使用while循环当que为空时停下来，每释放一个que中的root，将其加入list；将这个root的左节点和右节点再进去que；然后每结束一个循环再将list加入list0。

代码：

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> sec= new ArrayList<>();
        if(root == null){
            return sec;
        }
        Queue<TreeNode> que = new LinkedList<>();
        que.offer(root);
        while(!que.isEmpty()){
            int size = que.size();
            List<Integer> fir = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode node = que.remove();
                fir.add(node.val);
                if(node.left != null){
                    que.offer(node.left);
                }
                if(node.right != null){
                    que.offer(node.right);
                }
            }
            sec.add(fir);
        }
        return sec;
    }
}

链接：

二叉树的层序遍历

二从前序和中序遍历中构造二叉树

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

思路：写这种代码我们要知道：我们平时根据前序和中序遍历很容易推导出一个二叉树来，所以我们要做的就是将我们脑海中的过程转变成代码

首先前序遍历的第一个肯定是根节点，然后是先先往它的左节点走；但中序遍历的特点就是：在一个范围内，每个父节点的左边是它的左子树，而右边是它的右子树，所以我们要推断出这个范围；我们根据这两个特点就能够写出这个代码

比如：

根据preorder可知3是根节点，所以3的左边和右边分别是它的左子树和右子树

然后preorder的下一个是9，如果9在inorder的左边的话它就是3的左节点，相反就不是

代码：

class Solution {
    //i用于记录preorder的位置
    static int i;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        TreeNode root = new TreeNode();
        root = buildTreeChild(preorder, inorder, 0, preorder.length-1);
        //因为测试有多组测试，所以每次要将i置为0
        i = 0;
        return root;
    }
    private  TreeNode buildTreeChild(int[] preorder, int[] inorder, int ib, int ie){
        if(ib > ie || ib == -1){
            return null;
        }
        TreeNode root = new TreeNode(preorder[i]);
        //查找preorder[i]在inorder中的位置
        int pos = findIndex(inorder, preorder[i], ib, ie);
        i++;
        root.left = buildTreeChild(preorder, inorder, ib, pos-1);
        root.right = buildTreeChild(preorder, inorder, pos+1, ie);
        return root;
    }
    private int findIndex(int[] inorder, int val , int ib, int ie){
        int k= ib;
        for(k = ib; k <= ie; k++){
            if(val == inorder[k]){
                return k;
            }
        }
        //返回-2是因为在执行root.right时如果没有找到让ib=-1从而return null
        return -2;
    }
}

链接：

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

三中序和后序遍历构造二叉树

思路：

根据上一题我们已经有做这种题目的经验了，inorder还是根据范围来判断左子树和右子树。但postorder和preorder不一样的是它是最后一个节点是root，然后在postorder中往前走，根据范围来判断root的右子树的左子树。

根据postorder 3是根节点，inorder中3的左右分别是根节点的左子树和右子树

然后postorder往前走，20在3的右边，所以20是3的右节点

代码：

class Solution {
    static int i;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        TreeNode root = new TreeNode();
        i = postorder.length-1;
        root = buildTreeChild(inorder, postorder, 0, i);
        i = postorder.length-1;
        return root;
    }

    private  TreeNode buildTreeChild(int[] inorder, int[] postorder, int ib, int ie){
        if(ib > ie){
            return null;
        }
        TreeNode root = new TreeNode(postorder[i]);
        int pos = findIndex(inorder, postorder[i], ib, ie);
        i--;
        root.right = buildTreeChild(inorder, postorder, pos+1, ie);
        root.left = buildTreeChild(inorder, postorder, ib, pos-1);
        return root;
    }
    private int findIndex(int[] inorder, int val , int ib, int ie){
        int i = ib;
        for(i = ib; i <= ie; i++){
            if(val == inorder[i]){
                return i;
            }
        }
        return 0;
    }
}

链接：

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

注：题目均来自力扣