【数据结构】二叉树OJ题

相同的树

题目链接:LeetCode-100. 相同的树
题目描述:给你两棵二叉树的根节点 p 和 q ,编写一个函数来检验这两棵树是否相同。如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。

相同的树

代码:

public boolean isSameTree(TreeNode p, TreeNode q) {
    //如果p为空并且q不为空  或者  q为空p不为空,返回false
    if ((p == null && q != null) || (q == null && p != null)) {
        return false;
    }
    //两个为空  返回true
    if (p == null && q == null) {
        return true;
    }
    //两个不为空,但值不等,返回false
    if (p.val != q.val) {
        return false;
    }
    //根结点判断后,再判断左子树和右子树,只要有一个不同就返回false
    return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
    }

另一棵树的子树

题目链接:LeetCode-572. 另一棵树的子树
题目描述:给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

另一棵树的子树示例

代码:

public boolean isSubtree(TreeNode root, TreeNode subRoot) {
    if (root == null || subRoot == null) {
        return false;
    }
    //一个树是另一个树的子树
    //要么两棵树相同
    //要么是左树的子树 或 右树的子树
    if (isSameTree(root, subRoot))
        return true;
    //在判断是不是其左子树
    if (isSubtree(root.left, subRoot))
        return true;
    //在判断是不是其右子树
    if (isSubtree(root.right, subRoot))
        return true;
    return false;
}

//判断两棵树相不相同,上题代码写过
public boolean isSameTree(TreeNode p, TreeNode q) {
        //如果p为空并且q不为空  或者  q为空p不为空,返回false
        if ((p == null && q != null) || (q == null && p != null)) {
            return false;
        }
        //两个为空  返回true
        if (p == null && q == null) {
            return true;
        }
        //两个不为空,但值不等,返回false
        if (p.val != q.val) {
            return false;
        }
        return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
    }

另一棵树的子树实现过程

翻转二叉树

题目链接:LeetCode-226. 翻转二叉树
题目描述:给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。
翻转二叉树

实现过程:
实现过程
代码:

public TreeNode invertTree(TreeNode root) {
    if(root == null){
        return null;
    }
    //交换左右结点
    TreeNode tmp = root.right;
    root.right = root.left;
    root.left = tmp;
    //继续交换剩下的子树结点
    invertTree(root.left);
    invertTree(root.right);
    return root;
}

平衡二叉树

题目链接:LeetCode-110. 平衡二叉树
题目描述:给定一个二叉树,判断它是否是高度平衡的二叉树。本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

子树的高度差的超过 1 就不是平衡二叉树,所以我们可以计算出左右子树的高度并计算高度差。

代码:

public boolean isBalanced(TreeNode root) {
    if(root == null){
        return true;
    }
    //如果左右高度差大于1 就不是平衡二叉树
    if(Math.abs(getHeight(root.left) - getHeight(root.right)) > 1)
        return false;
    //左子树和右子树都要是平衡二叉树    
    return isBalanced(root.left) && isBalanced(root.right);
}

//求树的高度
public int getHeight(TreeNode root) {
    if (root == null) {
        return 0;
    }
    //求树的高度就是  先求出左子树的高度和右子树的高度,
    //谁高就返回谁的高度 最后在加上根结点的高度1,
    int leftHeight = getHeight(root.left);
    int rightHeight = getHeight(root.right);
    return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
}

使用上面的方法会使时间复杂度变高,我们可以改进 求树的高度 的方法,在求高度时,只要左右高度差大于 1,我们就返回 -1,避免重复计算。

public boolean isBalanced(TreeNode root) {
    return getHeight(root) >= 0;
}

public int getHeight(TreeNode root) {
    if (root == null) {
        return 0;
    }

    int leftHeight = getHeight2(root.left);
    int rightHeight = getHeight2(root.right);
    //计算出左右的高度后 直接判断左右高度差是否大于1
    if(leftHeight >= 0 && rightHeight >= 0 && Math.abs(leftHeight - rightHeight) <= 1){
        //高度差小于等于1  就返回左右高度较大值+1
        return leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
    }else
        //高度差大于1就不是平衡二叉树 直接返回-1
        return -1;  
}

对称二叉树

题目链接:LeetCode-101. 对称二叉树
题目描述:给你一个二叉树的根节点 root , 检查它是否轴对称

对称二叉树描述

对称二叉树实现过程
代码:

class Solution {
    public boolean isSymmetric(TreeNode root) {
        if(root == null){
            return true;
        }
        //传入根结点的左孩子和右孩子
        return compare(root.left, root.right);
    }

    public boolean compare(TreeNode leftChild, TreeNode rightChild) {
        //如果两个结点都为空
        if(leftChild == null && rightChild == null){
            return true;
        }
        //一个为空 一个不为空
        if(leftChild == null || rightChild == null){
            return false;
        }
        //都不为空 但值不相等
        if(leftChild.val != rightChild.val){
            return false;
        }
        //左树的左孩子和右树的右孩子相等  和   左树的右孩子和右树的左孩子相等 同时成立
        return compare(leftChild.left, rightChild.right) && compare(leftChild.right, rightChild.left);
    }
}

二叉树构建及遍历

题目链接:牛客网-KY11 二叉树遍历
题目描述:二叉树遍历描述
我们需要根据题目给出的前序遍历,构建出一颗二叉树,再根据二叉树输出中序遍历结果,如何构建出二叉树才是本题的关键。
如何建立二叉树

代码:

import java.util.Scanner;

class TreeNode {
    public char val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode(char val) {
        this.val = val;
    }
}

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        while (in.hasNextLine()) { // 注意 while 处理多个 case
            String str = in.nextLine();
            TreeNode root = creatTree(str);
            inOrder(root);
        }
    }
    //中序遍历
    public static void inOrder(TreeNode root){
        if(root == null){
            return;
        }
        inOrder(root.left);
        System.out.print(root.val+" ");
        inOrder(root.right);
    }
    //定义在外部,避免递归时 i又重0开始
    public static int i = 0;

    //根据字符串,创建二叉树
    public static TreeNode creatTree(String str) {
        TreeNode root = null;
        //如果字符不是#,就生成结点,继续递归
        if (str.charAt(i) != '#') {
            root = new TreeNode(str.charAt(i));
            //指向下一个字符
            i++;
            root.left = creatTree(str);
            root.right = creatTree(str);
        } else {
            //是#就指向下一字符
            i++;
        }
        //返回结点
        return root;
    }
}

二叉树的最近公共祖先

题目链接:LeetCode-236. 二叉树的最近公共祖先
题目描述:给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

最近公共祖先描述

几种情况:

几种情况

代码:

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root == null){
            return null;
        }
        //如果p 或q 等于根结点
        if(root == p || root == q){
            return root;
        }
        TreeNode leftRet = lowestCommonAncestor(root.left, p, q);
        TreeNode rightRet = lowestCommonAncestor(root.right, p, q);
        //都不为空  返回根结点
        if(leftRet != null && rightRet != null){
            return root;
            //其中一个为空
        }else if(leftRet != null){
            return leftRet;
        }else if(rightRet != null){
            return rightRet;
        }
        //都为空
        return null;
    }
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值