admin管理员组文章数量:1487745
【数据结构】翻转、平衡、对称二叉树,最大深度、判断两棵树是否相等、另一棵树的子树
检查两棵树是否相同
100. 相同的树 - 力扣(LeetCode)
思路解透
- 两个根节点一个为空一个不为空的话,这两棵树就一定不一样了
- 若两个跟节点都为空,则这两棵树一样
- 当两个节点都不为空时:
- 若两个根节点的值不相同,则这两棵树不一样
- 若两个跟节点的值相同,则对左右两棵子树进行递归判断
代码解析
代码语言:javascript代码运行次数:0运行复制/**
* 时间复杂度为:O(min(m,n))
* @param p m个节点
* @param q n个节点
* @return
*/
public boolean isSameTree(TreeNode p, TreeNode q) {
//1. 一个为空,一个不为空,必不一样
if(p == null && q != null || p != null && q == null){
return false;
}
//2. 两个都为空
if(p == null && q == null){
return true;
}
//3. 剩下的一种情况就是两个都不为空,不需要再用if限制条件了
if(p.val != q.val){
return false;
}
//4. 此时代表两个都不为空,且 val 的值相等
//5. 说明根节点相同,系接下来判断两棵树的左右是不是同时分别相同
return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);
}
另一棵树的子树
572. 另一棵树的子树 - 力扣(LeetCode)
思路解透
注意: 当两棵树相同时,也返回 true
- 首先判断两棵树是否相同,若相同,返回
true
(需要调用上面一题的方法) - 若不相同,判断是否是左子树的子树,是否是右子树的子树
- 若都不是,则返回
false
代码解析
代码语言:javascript代码运行次数:0运行复制/**
* 判断两棵树是否相同
* @param p
* @param q
* @return
*/
public boolean isSameTree(TreeNode p, TreeNode q){
if(p == null && q == null){
return true;
}
if(p != null && q == null || p == null && q != null){
return false;
}
//都不为空
if(p.val != q.val){
return false;
}
//对子树进行判断
return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);
}
/**
* 判断是不是子树
* 时间复杂度:O(m*n)
* @param root
* @param subRoot
* @return
*/
public boolean isSubtree(TreeNode root, TreeNode subRoot) {
if(root == null){
return false;
}
//1. 两棵树若相同
if(isSameTree(root,subRoot)){
return true;
}
//2. 判断是否是左子树的子树
if(isSubtree(root.left,subRoot)){
return true;
}
//3. 判断是否是右子树的子树
if(isSubtree(root.right,subRoot)){
return true;
}
return false;
}
翻转二叉树
226. 翻转二叉树 - 力扣(LeetCode)
思路解透
- 若跟节点为空就返回
null
- (优化步骤)若左右两边都为空,就不需要交换了,直接返回
root
- 定义一个
ret
节点作为中间人,将左右子节点进行交换 - 递归对左右子节点的左右子节点进行交换
- 返回
root
代码解析
代码语言:javascript代码运行次数:0运行复制/**
* 翻转二叉树
* @param root
* @return
*/
public TreeNode invertTree(TreeNode root) {
if(root == null) {
return null;
}
if(root.left == null && root.right == null){
return root;
}
//左右子节点进行交换
TreeNode ret = root.right;
root.right = root.left;
root.left = ret;
invertTree(root.left);
invertTree(root.right);
return root;
}
二叉树最大深度
104. 二叉树的最大深度 - 力扣(LeetCode)
思路解透
root
下来之后,每次都是取左右两边更高的那一个,再+1
递归上去
代码解析
代码语言:javascript代码运行次数:0运行复制//获取二叉树的高度
public int maxDepth(TreeNode root) {
if(root == null){
return 0;
}
int leftDepth = maxDepth(root.left);
int rightDepth = maxDepth(root.right);
return leftDepth > rightDepth ? leftDepth + 1 : rightDepth + 1;
}
平衡二叉树
110. 平衡二叉树 - 力扣(LeetCode)
思路解透
平衡二叉树: 所有节点的左右子树高度差小于等于 1
- 当前
root
的左右子树高度差小于等于1
- 用到
Math.abs()
方法,得到的是() 里面的绝对值
- 用到
- 同时满足
root
的左子树平衡&&
root
的右子树平衡
代码解析
代码语言:javascript代码运行次数:0运行复制/**
* 获取最大深度
* @param root
* @return
*/
public int maxDepth(TreeNode root) {
if(root == null){
return 0;
}
int leftDepth = maxDepth(root.left);
int rightDepth = maxDepth(root.right);
return leftDepth > rightDepth ? leftDepth + 1 : rightDepth + 1;
}
/**
* 平衡二叉树
* 时间复杂度为:O(n^2)
* @param root
* @return
*/
public boolean isBalanced(TreeNode root) {
if(root == null){
return true;
}
int leftDepth = maxDepth(root.left);
int rightDepth = maxDepth(root.right);
if(Math.abs(leftDepth - rightDepth) <= 1 &&
isBalanced(root.left) && isBalanced(root.right)){
return true;
}
return false;
}
代码优化(字节笔试)
在时间复杂度为 O(n) 的条件下,完成平衡二叉树的判断
若要让时间复杂度为O(n)
,则需要在判断的过程中,只要发现左右俩树高度相差大于 1
,就直接 return -1
,不再进行后续判断了
/**
* 获取最大深度
* @param root
* @return
*/
public int maxDepth(TreeNode root) {
if(root == null){
return 0;
}
int leftDepth = maxDepth(root.left);
if(leftDepth < 0)
return -1;
int rightDepth = maxDepth(root.right);
if(rightDepth < 0)
return -1;
if(Math.abs(leftDepth - rightDepth) <= 1){
return Math.max(leftDepth, rightDepth);
}else {
return -1;
}
}
/**
* 平衡二叉树
* 时间复杂度为:O(n^2)
* @param root
* @return
*/
public boolean isBalanced(TreeNode root) {
if(root == null){
return true;
}
return maxDepth(root) >= 1;
}
对称二叉树
101. 对称二叉树 - 力扣(LeetCode)
思路解透
需要判断 root
左树和右树是否对称
p
的左树和q
的右树是否对称p
的右树和q
的左树是否对称
- 结构
- 一个为空,一个不为空
- 两个都为空
- 两个都不为空
- 值:建立在两个引用都不为空的情况下,判断
val
代码解析
代码语言:javascript代码运行次数:0运行复制public boolean isSymmetric(TreeNode root) {
if (root == null) return true;
return isSymmetricChild(root.left, root.right);
}
public boolean isSymmetricChild(TreeNode leftTree, TreeNode rightTree) {
//1. 检查结构是否相同
//1.1 一个为空一个不为空
if (leftTree != null && rightTree == null || leftTree == null && rightTree != null) {
return false;
}
//1.2 处理两个都为空和两个都不空的情况
if (leftTree == null && rightTree == null) {
return true;
}
//1.3 两个都不为空,判断他们的值一不一样
if (leftTree.val != rightTree.val) {
return false;
}
//此时两个节点都不为空,且值一样
//2. 开始判断是否对称,需要满足
// 左子树的左 和 右子树的右对称 通同时 左子树的右 和 右子树的左对称
return isSymmetricChild(leftTree.left, rightTree.right) && isSymmetricChild(leftTree.right, rightTree.left);
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。 原始发表:2024-09-04,如有侵权请联系 cloudcommunity@tencent 删除二叉树数据结构nullreturnroot本文标签: 数据结构翻转平衡对称二叉树,最大深度判断两棵树是否相等另一棵树的子树
版权声明:本文标题:【数据结构】翻转、平衡、对称二叉树,最大深度、判断两棵树是否相等、另一棵树的子树 内容由网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.betaflare.com/shuma/1754845957a3180328.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论