题目描述:
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
给定的树 B:
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
链接:https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
思路:找寻子结构,分为两种情况,该子结构在左边或者在右边。则进行递归判断
- 找到了两值相等节点,开始检测是否是子结构
- 不是,再继续递归,判断后续的子结构
- 是,直接返回true
- 最后返回三者的判断情况。
代码:
class Solution {
public:
bool IsSubTree(TreeNode* A, TreeNode* B)//检测子结构
{
if(B == nullptr)//B为空了,肯定是子结构
return true;
if(A == nullptr)//A为空了,肯定不是子结构
return false;
if(A->val != B->val)//A 和 B值不相等了,肯定不是
return false;
return IsSubTree(A->left,B->left) && IsSubTree(A->right,B->right);//左右两边同时判断
}
bool isSubStructure(TreeNode* A, TreeNode* B) {
if(A == nullptr || B == nullptr)//若是有一个为空,直接返回false
return false;
bool cur = false;
if(A->val == B->val)//有相等了,检测是否为子结构
cur = IsSubTree(A,B);
bool left = isSubStructure(A->left,B);//左边是否是子结构
bool right = isSubStructure(A->right,B);//右边是否是子结构
return cur || left || right;//返回这
}
};