输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路:递归(涉及到遍历树结构)
主函数:
边界条件:if A==null or B==null:false
else:
if A->val==B->val:以此为节点开始进行比对(记为函数isSubtree)
else if:
判断A的左子树
else:
判断A的右子树
isSubtree:
边界条件:if B==null:true;//B遍历结束没有发现问题
if A==null:false;//B未结束,而A结束了
if A->val==B->val:
判断左子树;
if ->left相同:
判断右子树
都相同 B是A的子结构(true)
其中第一次书写过程中出现的问题:
1、B是A的子结构,不代表对于B的每一个节点,都有B==A
例如:B只包含根节点,左右子树为空的情况下,只要有A->val==B->val即可,而不需要A的该节点左右子树均为null
所以,在进行判断时,只需要判断a->val==b->val即可
2、两个函数(主函数和子函数)中,均有判断左右子树的过程,但是他们的含义是不同的
在主函数中:在判断A的根节点不能与B根节点匹配时,会判断A的左右子树是否与B节点匹配,所以传入参数为A->left(right)与B
在子函数中:在判断A的根节点可与B的根节点匹配时,将继续判断各自的左右子树是否匹配,所以传入参数为
A->left(right)与B->left(right)
通过代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
bool isSubtree(TreeNode* p1, TreeNode* p2)
{
if(p2==NULL) return true;
if(p1==NULL) return false;
if(p1->val==p2->val)
return isSubtree(p1->left, p2->left) && isSubtree(p1->right, p2->right);
else return false;
}
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool res=false;
if(pRoot1==NULL || pRoot2==NULL)
return false;
if(pRoot1->val==pRoot2->val)
res=isSubtree(pRoot1,pRoot2);
if(!res) res=HasSubtree(pRoot1->left, pRoot2);
if(!res) res=HasSubtree(pRoot1->right,pRoot2);
return res;
}
};