剑指offer题目17:树的子结构
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解答思路
画图出来先分析思路。

- 先判断两者的根是否相同
- 相同则判断两者的左右子树
- 不相同则用父树的左子树去和子树比对
- 如果子树已经到达终点,则结束比对
分析完了会觉得不算太难,觉得就是难在判断边界条件,注意某一方为NULL的时候。
实现代码
/*
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)
{
bool result = false;
if(pRoot1 != NULL && pRoot2 != NULL) {
// 如果两者根相等,判断左右子树
if(pRoot1->val == pRoot2->val) {
result = ConfirmSubtree(pRoot1, pRoot2);
}
// 如果不相等,则用父树的左树和子树比较
if(!result) {
result = HasSubtree(pRoot1->left, pRoot2);
}
// 如果不相等,则用父树的右树和子树比较
if(!result) {
result = HasSubtree(pRoot1->right, pRoot2);
}
}
return result;
}
// 在知道根节点相同的情况下,判断左右子树。
// 递归的结束条件是两者都到达了最底部。
bool ConfirmSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {
// 先判断子树是否为NULL,这是因为如果子树和父树都为NULL的话,其实就是子树包含在父树里面
if(pRoot2 == NULL) {
return true;
}
if(pRoot1 == NULL) {
return false;
}
if(pRoot1->val != pRoot2->val) {
return false;
}
return ConfirmSubtree(pRoot1->left, pRoot2->left) && ConfirmSubtree(pRoot1->right, pRoot2->right);
}
};
作者:大傻逼
本文版权归作者所有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

浙公网安备 33010602011771号