题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路
举个例子,如图。可以知道 B 树是 A 树的子结构。在判断的过程中,可以分为两步。第一步,在 A 树中找到和 B 树的根结点值一样的结点 R;第二步,判断 A 树中以结点 R 为根的子树是不是包含了 B 树,即左右结点 (左右子树) 和 B 树一样。
第一步,要在 A 树中找到和 B 树根节点值相同的结点 R,就要遍历 A 树。二叉树的遍历往往用递归方法实现,那么我们采用递归的方法。找到递归两要素,结束条件是找到与 B 树根结点值相同的结点,即 treeB->val == treeA->val;递归公式是先从根判断,然后是左子树,再是右子树,也就是先序遍历。
第二步,要判断以 R 为结点的子树是否包含 B 树,其实质也是树的遍历。我们先考虑递归结束条件,分别是 B 树递归结点为空,A 树递归结点为空和 B 树递归结点值与 A 树递归结点值不同。而递归公式则是 A、B 树左子树的判断和右子树的判断。
实现代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(!pRoot1||!pRoot2) return false;
if(pRoot1->val == pRoot2->val){
if(IsSubtree(pRoot1, pRoot2))
return true;
}
return HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
}
bool IsSubtree(TreeNode* p1, TreeNode* p2){
// 判断 A B树为空的顺序不可以换。
// 如果先判断 A 为空,此时 B 树也可能为空,
// 两者同时结束叶子结点的遍历,这时应该返回 true。
if(!p2) return true;
if(!p1) return false;
if(p1->val != p2->val) return false;
return IsSubtree(p1->left, p2->left) && IsSubtree(p1->right, p2->right);
}
};
注意:
-
与二叉树相关的代码有大量的指针操作,注意考虑空指针nullptr。
-
计算机表示小数(float和double型小数)都有误差,判断两个小数是否相等时,不能直接用等号(==),而是用两个小数的差的绝对值是否很小进行判断,例如10^(-7),num1-num2>10^(-7),num1-num2<10^(-7)
参考链接:
1.https://www.nowcoder.com/profile/2019821/codeBookDetail?submissionId=14071376