作者:~小明学编程
文章专栏:每日一练——Java
格言:目之所及皆为回忆,心之所想皆为过往
目录
相同的树
链接:
题目描述
给你两棵二叉树的根节点 p 和 q ,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入:p = [1,2,3], q = [1,2,3]
输出:true
示例 2:
输入:p = [1,2], q = [1,null,2]
输出:false
示例 3:
输入:p = [1,2,1], q = [1,1,2]
输出:false
提示:
两棵树上的节点数目都在范围 [0, 100] 内
-104 <= Node.val <= 104
代码
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
if (p!=null && q!=null) {
if (p.val==q.val) {
return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);
} else {
return false;
}
} else if (p==null && q==null) {
return true;
} else {
return false;
}
}
}
此题想通了就非常简单,但是我在写的时候还是犹豫了一下,犹豫的原因就是第五行代码,我在思考返回值怎么写,刚开始想的是肯定不能一一的返回左和右直接调用也不行,同时第十行代码也不知道该如何去返回了,然后看了一眼评论区,恍然大悟,我们可以&&与逻辑解决这个问题,这样一并解决了第十行的问题,当左边都为空时我们就可以大胆的返回true了,因为有一个与&&逻辑,我们还要去判断右边的树的结构以及值,完美的解决了这个问题。
另一颗树
链接:
题目描述
给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。
二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。
示例 1:
输入:root = [3,4,5,1,2], subRoot = [4,1,2]
输出:true
示例 2:
输入:root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
输出:false
提示:
root 树上的节点数量范围是 [1, 2000]
subRoot 树上的节点数量范围是 [1, 1000]
-104 <= root.val <= 104
-104 <= subRoot.val <= 104
代码
class Solution {
private boolean isSameTree(TreeNode p, TreeNode q) {
if (p!=null && q!=null) {
if (p.val==q.val) {
return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);
} else {
return false;
}
} else if (p==null && q==null) {
return true;
} else {
return false;
}
}
public boolean isSubtree(TreeNode root, TreeNode subRoot) {
if (root!=null ) {
return isSameTree(root,subRoot) || isSubtree(root.left,subRoot) || isSubtree(root.right,subRoot);
} else {
return false;
}
}
}
这题的难点还是返回值,想到了就很简单,想不到就很难,判断相等我们用前面一题的代码,这里主要是这个或者的用法当我们判断出当前节点不相等的时候我们得,递归自身分别调用左子树或者右子树。
平衡二叉树
链接:
题目描述
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
示例 3:
输入:root = []
输出:true
提示:
树中的节点数在范围 [0, 5000] 内
-104 <= Node.val <= 104
代码
//方法一
// class Solution {
// private int height(TreeNode root) {
// if (root == null) {
// return 0;
// } else {
// int leftHeight = height(root.left) + 1;
// int rightHeight = height(root.right) + 1;
// return Math.max(leftHeight,rightHeight);
// }
// }
// public boolean isBalanced(TreeNode root) {
// if (root==null) {
// return true;
// }
// int leftHeight = height(root.left);
// int rightHeight = height(root.right);
// return Math.abs(leftHeight-rightHeight)<=1 && isBalanced(root.left) && isBalanced(root.right);
// }
// }
//方法二
class Solution {
private int height(TreeNode root) {
if (root == null) {
return 0;
} else {
int leftHeight = height(root.left) + 1;
if (leftHeight==0) return -1;
int rightHeight = height(root.right) + 1;
if (rightHeight==0) return -1;
if (Math.abs(leftHeight-rightHeight) <= 1) {
return Math.max(leftHeight,rightHeight);
} else {
return -1;
}
}
}
public boolean isBalanced(TreeNode root) {
if (root == null) {
return true;
}
if (height(root)==-1) {
return false;
} else {
return true;
}
}
}
此题前后给了两种方法,第一种方法的时间复杂度过大,不建议这样写,第二种的时间复杂度只有O(n)。