大致思路:
因为你可以包含在树上的很多地方,所以必须要先找到这个地方——先找到A的子树中和B树具有共同根结点的地方。那肯定就是要遍历,在二叉树上就是递归咯。
找到共同根节点后,再判断是否“包含”。之前写过如何判断两棵树是否相同的代码(递归判断),这里只需要改下判断条件就行。判断条件:如果a为空b也为空,或者a不为空b为空,都可以说明b包含于a,tree。而如果a空而b不空,说明a并没有把b包含完全啊,所以就false。
AC代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool HasIt(TreeNode* a, TreeNode* b) //判断a是否包含b
{
if((a==NULL && b==NULL) || (a!=NULL && b==NULL)) //这两种情况一种是相等一种是包含,都可以
return true;
if(a==NULL && b!=NULL)
return false;
if(a->val!=b->val)
return false;
return HasIt(a->left,b->left) && HasIt(a->right,b->right);
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot2==NULL) //先判空,同时也为递归作为临界条件
return false;
if(pRoot1==NULL)
return false;
//先以当前为共同根节点,若不相同则以孩子结点为共同根节点(因为若要包含,肯定proot1的某个子树和proot2共同根节点)
bool result = HasIt(pRoot1,pRoot2);
if(result==true)
return true;
else
return HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2); //这里很巧妙
}
};