二叉树oj题集(LeetCode)

简介: 二叉树oj题集(LeetCode)

100. 相同的树

关于树的递归问题,永远考虑两方面:返回条件和子问题

  1. 先考虑返回条件,如果当前的根节点不相同,那就返回false(注意,不要判断相等时返回什么,因为当前相等并不能说明后面节点相等,所以要转换为不相等返回什么)
  2. 但是还要考虑为空的情况,如果两个树的根节点都为空,则返回true(只有经过层层比较,当比较到最后都为空树时,才返回true);如果一个为空一个不为空,则返回false
  3. 最后考虑子问题,当前树的根节点比较完毕,那就转化为左子树和右子树进行递归比较
bool isSameTree(struct TreeNode* p, struct TreeNode* q)
{
    if (p == NULL && q == NULL)
    {
        return true;
    }
 
    if (p == NULL || q == NULL)
    {
        return false;
    }
 
    if (p->val != q->val)
    {
        return false;
    }
 
    return isSameTree(p->left, q->left)
        && isSameTree(p->right, q->right);
}

具体方法:

  • 如果为空树,则返回true(必然走到末尾,那前面的值判断都通过)
  • 当前节点与左孩子的值进行比较,如果不相等,则返回false(注意,加上条件判断,保证左孩子不为空,防止对空指针的解引用)
  • 同理,当前节点与右孩子的值进行比较,如果不相等,则返回false
  • 最后的子问题,则返回左子树与右子树的返回值的逻辑与,只要不满足上述条件,就一直往下递归
bool isUnivalTree(struct TreeNode* root)
{
    if (root == NULL)
    {
        return true;
    }
 
    if (root->left && root->val != root->left->val)
    {
        return false;
    }
 
    if (root->right && root->val != root->right->val)
    {
        return false;
    }
 
    return isUnivalTree(root->left)
       && isUnivalTree(root->right);
}

101. 对称二叉树

思路:因为要用左子树的左支(右支)和右子树的右支(左支)进行比较,所以再创建一个子函数接受两个参数。

具体方法:

  1. 如果左右子树根节点都为NULL,则返回true
  2. 如果左右子树根节点一个为空一个不为空,则返回false
  3. 如果左右子树根节点非空,并且不相等,则返回false
  4. 子问题划分为,左子树的左支与右子树的右支相比,左子树的右支与右子树的左支相比,检查是否对称,如果都对称,则返回true
bool _isSymmetric(struct TreeNode* left, struct TreeNode* right)
{
    if (left == NULL && right == NULL)
    {
        return true;
    }
 
    if (left == NULL || right == NULL)
    {
        return false;
    }
 
    if (left->val != right->val)
    {
        return false;
    }
 
    return _isSymmetric(left->left, right->right)
        && _isSymmetric(left->right, right->left);
}
 
bool isSymmetric(struct TreeNode* root)
{
    return _isSymmetric(root->left, root->right);
}

首先,这题要注意的是,它要求动态开辟一个数组,将二叉树的值以前序遍历的顺序放入,最后再返回该数组,所以它跟我们之前直接前序遍历还是不一样的,变得更加复杂了。

具体实现思路:

  1. 这道题,因为数组大小不确定,所以题目传进来一个返回型参数(returnSize),那么我们则把求二叉树的节点个数的函数写出来(对现在的我们来说,应该挺简单的),再将节点个数求出,赋给returnSize
  2. 动态开辟数组,大小为二叉树节点个数个整型空间
  3. 创建前序遍历子函数,用来真正实现以前序遍历的顺序,将二叉树的值放入数组
  4. 注意,这里如果传进来i,那么会因为其为局部变量,递归时内层函数栈帧i++,并不会影响到外层函数栈帧中的i,导致数组中有些值被重叠覆盖,而有些值未初始化(随机值)。所以,正确的做法,是传入i的地址,对其解引用++

传入i时,会造成内存错误,如下图:

void _preorderTraversal(struct TreeNode* root, int* a, int* pi)
{
    if (root == NULL)
    {
        return;
    }
 
    a[(*pi)++] = root->val;
    _preorderTraversal(root->left, a, pi);
    _preorderTraversal(root->right, a, pi);
}
 
int BTreeSize(struct TreeNode* root)
{
    return root == NULL ? 0 : BTreeSize(root->left) + BTreeSize(root->right) + 1;
}
 
int* preorderTraversal(struct TreeNode* root, int* returnSize)
{
    *returnSize = BTreeSize(root);
    int* a = (int*)malloc(*returnSize * sizeof(int));
 
    int i = 0;
    _preorderTraversal(root, a, &i);
 
    return a;
}

572. 另一棵树的子树

思路:这题可以复用 100. 相同的树 这题的代码,来转化为比较root树的任意一棵子树,是否存在与subRoot树相等,如果相等,则代表是它的子树

具体方法:

  1. 将isSameTree的判断代码复用过来
  2. 返回条件:当前root子树与subRoot树相等时,返回true(注意,判断不相等没有意义,因为当前不相等,可以递归到下一层找相等)
  3. 子问题:如果当前树本身不等于subRoot树,则转化为左子树和右子树与subRoot树判断是否相等,只要有一边相等,则返回true
  4. 返回条件:注意到递归下去,root有可能为空指针,所以加上如果root为NULL时,返回false(因为题目规定subRoot一定不为空,所以子树为空则一定不相等),同时防止递归时对空指针的解引用
bool isSameTree(struct TreeNode* p, struct TreeNode* q)
{
    if (p == NULL && q == NULL)
    {
        return true;
    }
 
    if (p == NULL || q == NULL)
    {
        return false;
    }
 
    if (p->val != q->val)
    {
        return false;
    }
 
    return isSameTree(p->left, q->left)
        && isSameTree(p->right, q->right);
}
 
bool isSubtree(struct TreeNode* root, struct TreeNode* subRoot)
{
    if (root == NULL)
    {
        return false;
    }
 
    if (isSameTree(root, subRoot))
    {
        return true;
    }
 
    return isSubtree(root->left, subRoot)
       || isSubtree(root->right, subRoot);
}


相关文章
|
2月前
【LeetCode 31】104.二叉树的最大深度
【LeetCode 31】104.二叉树的最大深度
28 2
|
2月前
【LeetCode 29】226.反转二叉树
【LeetCode 29】226.反转二叉树
21 2
|
2月前
【LeetCode 28】102.二叉树的层序遍历
【LeetCode 28】102.二叉树的层序遍历
19 2
|
2月前
【LeetCode 43】236.二叉树的最近公共祖先
【LeetCode 43】236.二叉树的最近公共祖先
22 0
|
2月前
【LeetCode 38】617.合并二叉树
【LeetCode 38】617.合并二叉树
20 0
|
2月前
【LeetCode 37】106.从中序与后序遍历构造二叉树
【LeetCode 37】106.从中序与后序遍历构造二叉树
25 0
|
2月前
【LeetCode 34】257.二叉树的所有路径
【LeetCode 34】257.二叉树的所有路径
23 0
|
2月前
【LeetCode 32】111.二叉树的最小深度
【LeetCode 32】111.二叉树的最小深度
19 0
|
4月前
|
存储 算法
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
文章深入探讨了二叉树的层序遍历方法,并展示了如何通过队列实现层序遍历的算法逻辑,同时指出掌握层序遍历技巧可以帮助解决LeetCode上的多道相关题目。
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
|
4月前
|
算法 Java
LeetCode第94题二叉树的中序遍历
文章介绍了LeetCode第94题"二叉树的中序遍历"的解法,使用递归实现了中序遍历的过程,遵循了"左根右"的遍历顺序,并提供了清晰的Java代码实现。
LeetCode第94题二叉树的中序遍历