【刷题笔记】572. 另一棵树的子树

130 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第20天,点击查看活动详情

一、题目描述:

572. 另一棵树的子树 - 力扣(LeetCode) (leetcode-cn.com)

给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

 

示例 1:

image.png

输入:root = [3,4,5,1,2], subRoot = [4,1,2]
输出:true

示例 2:

image.png

输入:root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
输出:false

提示:

  • root 树上的节点数量范围是 [1, 2000]
  • subRoot 树上的节点数量范围是 [1, 1000]
  • -10^4 <= root.val <= 10^4
  • -10^4 <= subRoot.val <= 1064

二、思路分析:

  • 对称性问题主要有两类
  • 一类是都从根节点开始判断(此时只用一个递归即可实现,但是要注意是子树还是树的子结构);
  • 第二类是从整个树来判断,即不一定只从根节点开始,因此需要两个递归来完成,其中第一个递归就是从根节点开始来进行判断(类似于一类的答案),第二个递归则要以不同节点作为根节点依次进行判断。
  • 这个题的前半部分递归就是100. 相同的树,该递归就是判断一个树从根节点开始是否是相同的,本题是求子树,其实就是不断移动root的节点,以root的其他节点作为新的根节点,并继续进行判断看是否有相同的部分,如果有则说明subRoot是root的一个子树。

三、AC 代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    //从根节点开始判断是否是一个子树
    bool isSameRTree(TreeNode* root, TreeNode* subRoot){
        if(root == nullptr && subRoot == nullptr){
            return true;
        }
        else if(root && subRoot == nullptr){
            return false;
        }
        else if(root == nullptr && subRoot){
            return false;
        }
        else if(root->val != subRoot->val){
            return false;
        }
        return isSameRTree(root->left,subRoot->left) && isSameRTree(root->right,subRoot->right);
    }
    bool isSubtree(TreeNode* root, TreeNode* subRoot) {
        if(root == nullptr){
            //主树为空了,子树还没匹配成功
            return false;
        }
        if(subRoot == nullptr){
            //题目说了subRoot不为空
        }

        //判断以当前节点为根节点时,subroot是否是root的子树
        if(isSameRTree(root,subRoot)){
            return true;
        }
        return isSubtree(root->left,subRoot) || isSubtree(root->right,subRoot);

    }
};