LeetCode 572. 另一棵树的子树 | C++语言版
LeetCode 572. 另一棵树的子树
题目描述
题目地址:572. 另一棵树的子树
给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。
二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。
解题思路
思路一:使用栈
代码实现
C++
/**
* 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 check(TreeNode* r,TreeNode* st){
//两树都为空
if(!r && !st) return true;
//两树有一个为空或者节点值不一样
if((r && !st) ||(!r && st) || (r->val!=st->val)) return false;
//比较两树是否相同
return check(r->left,st->left) && check(r->right,st->right);
}
//前序遍历二叉树
bool dfs(TreeNode* r, TreeNode* st) {
if(!r) return false;
//中左右
return check(r,st) || dfs(r->left,st) || dfs(r->right,st);
}
bool isSubtree(TreeNode* root, TreeNode* subRoot) {
//判断一棵树subRoot是不是另一颗树root的子树
//让两个指针一开始先指向root和subRoot的根,然后「同步移动」两根指针来「同步遍历」这两棵树,判断对应位置是否相等。
return dfs(root,subRoot);
}
};
运行结果
参考文章:
思路二:减少遍历节点数
代码实现
C++
在这里插入代码片
运行结果
参考文章:
LeetCode 剑指 Offer 26. 树的子结构
题目描述
题目地址:剑指 Offer 26. 树的子结构
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/
1 2
给定的树 B:
4
/
1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
解题思路
思路一:使用栈
代码实现
C++
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
private:
//判断B是否为A的子结构,其中B子结构的起点为A的根节点
bool recur(TreeNode* A, TreeNode* B) {
// 若B走完了,说明查找完毕,B为A的子结构
if(B == NULL) return true;
// 若B不为空并且A为空或者A与B的值不相等,直接可以判断B不是A的子结构
if(A == NULL || A->val != B->val) {
return false;
}
// 当A与B当前节点值相等,若要判断B为A的子结构
// 还需要判断B的左子树是否为A左子树的子结构 && B的右子树是否为A右子树的子结构
// 若两者都满足就说明B是A的子结构,并且该子结构以A根节点为起点
return recur(A->left, B->left) && recur(A->right, B->right);
}
public:
//判断B是否为A的子结构
bool isSubStructure(TreeNode* A, TreeNode* B) {
// 若A与B其中一个为空,立即返回false
if(A == NULL || B == NULL) {
return false;
}
// B为A的子结构有3种情况,满足任意一种即可:
// 1.B的子结构起点为A的根节点,此时结果为recur(A,B)
// 2.B的子结构起点隐藏在A的左子树中,而不是直接为A的根节点,此时结果为isSubStructure(A.left, B)
// 3.B的子结构起点隐藏在A的右子树中,此时结果为isSubStructure(A.right, B)
return recur(A, B) || isSubStructure(A->left, B) || isSubStructure(A->right, B);
}
};
运行结果
参考文章:
思路二:减少遍历节点数
代码实现
C++
在这里插入代码片