相同的树
题目链接: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;
}
}