携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第20天,点击查看活动详情
一、题目描述:
572. 另一棵树的子树 - 力扣(LeetCode) (leetcode-cn.com)
给你两棵二叉树 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]
- -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);
}
};