【剑指offer】树的子结构
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路
递归要有递归的亚子,嗯~这道题果然是递归的亚子,要理清哪里需要递归,首先从A的根节点开始,先找到与B的根节点相同的节点,加入A中C节点与B的根节点相同,然后开始判断C节点的左子树、右子树和B节点的左子树、右子树是否相同,都满足之后返回true。
代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool judge(TreeNode* A, TreeNode* B){
if(!B) return true;
if(!A) return false;
if(A->val == B->val){
return judge(A->left,B->left) && judge(A->right,B->right);
}
return false;
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(!pRoot1 || !pRoot2){
return false;
}
return judge(pRoot1,pRoot2) ||
HasSubtree(pRoot1->left,pRoot2) || //这是在A的左子树的地方找有无跟B节点相同的节点
HasSubtree(pRoot1->right,pRoot2); //若左子树没有就找右子树
}
};