剑指Offer_编程题
树的子结构
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
//树结点定义
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
//判断子树
bool isChildTree(TreeNode* A, TreeNode* B){
if(B == NULL)
return true;
if(A == NULL)
return false;
if(A->val == B->val){
return (isChildTree(A->left,B->left) && isChildTree(A->right,B->right));
}
return false;
}
//判断子结构
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if( pRoot1==NULL || pRoot2 == NULL)
return false;
return (isChildTree(pRoot1,pRoot2) || HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2));
}
分析:
树的子结构:子树的所有结点都出现在主树上;
递归解决:二叉树结构固定,都是根和左右结点,使用递归的方法遍历每一个结点,遍历的顺序为:根、左、右
- 判断以树A以当前结点为根节点时,树B是否为树A的子树;
- 递归遍历树A的左子树;
- 递归遍历树A的右子树;
一旦判断子树时返回的结果为true,即B是A的子结构,结束程序;否则,遍历到A的最后一个结点,返回flase;