JZ17_树的子结构
知识点:二叉树
题目链接
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
示例1
输入: {8,8,#,9,#,2,#,5},{8,9,#,2}
返回值: true
解题思路
- 这是由两个小问题组成的一道题
- 首先我们要判断一棵树是否和另一棵树有相同结构,这里需要用到递归
- 递归中最重要的是返回条件:如果B为空 则返回true;如果A为空 B不为空 则返回false;
- 如果A树的结点值 == B树的结点值,两个树左子树和右子树 递归进行下去
- 现在我们已经知道如何判断了,那么如何遍历树的结点呢,可以采用二叉树的遍历方式(前序、中序、后序都行)
void preOrder(TreeNode *r) { if (!r) return; // process r preOrder(r->left); preOrder(r->right); }
代码
#include "cheader.h"
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 ans = false;
if(pRoot1 == nullptr || pRoot2 == nullptr)
return false;
if(pRoot1->val == pRoot2->val)
ans = check(pRoot1, pRoot2);
if(ans == false)
ans = HasSubtree(pRoot1->left, pRoot2);
if(ans == false)
ans = HasSubtree(pRoot1->right, pRoot2);
return ans;
}
bool check(TreeNode* a,TreeNode* b){
if(a == nullptr && b != nullptr)
return false;
if(b == nullptr)
return true;
if(a->val == b->val)
return check(a->left,b->left)&&check(a->right,b->right);
else
return false;
}
};
int main()
{
Solution s;
TreeNode* head1 = new TreeNode(1);TreeNode* node1 = new TreeNode(2);TreeNode* node2 = new TreeNode(3);
head1->left = node1;head1->right = node2;
TreeNode* head2 = new TreeNode(1);TreeNode* node3 = new TreeNode(2);
head2->left = node3;
cout<<s.check(head1, head2)<<endl;
return 0;
}
今天也是爱zz的一天!