-
题目描述(from nowcoder):
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构) -
分析:
树1是原树,树2是要被匹配的树。
思路总体分为两部分,第三部分是对特殊情况的处理:
- 遍历树1找与树2根值相同的节点,找打了之后就继续判断其结构是否相匹配;
- 如果不匹配,则返回到开始判断的节点,继续遍历树1的左子树与右子树重复步骤1的做法,直到树2被遍历完成之后都和树1的某一部分匹配为止返回true,否则返回false。
- 整体实现思路完成之后,考虑特殊情况,比如两个树是否为空,要进行特殊处理;还有当树1被遍历完成还没有匹配上树2的时候,也返回false。
注意: 写遍历代码时,高度警惕每一处的空指针,对其处理,否则会造成整个程序崩溃。
- 代码:
/*
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* pRoot1, TreeNode* pRoot2)
{
if(pRoot2==nullptr)//树2被匹配完了,则返回true
return true;
if(pRoot1 == nullptr)//树1遍历完了但树2还没匹配完,返回false
return false;
if(pRoot1->val != pRoot2->val)
return false;
return Judge(pRoot1->left, pRoot2->left)&&Judge(pRoot1->right, pRoot2->right);
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool r = false;
//特殊处理
if( pRoot1==nullptr || pRoot2==nullptr)
return r;
else
{
if(pRoot1->val == pRoot2->val)//找到一个根节点相同地方,则继续判断结构是否完全匹配
r = Judge(pRoot1, pRoot2);
//否则继续匹配树1的左子树与右子树
if(!r)
r = HasSubtree(pRoot1->left, pRoot2);
if(!r)
r = HasSubtree(pRoot1->right, pRoot2);
}
return r;
}
};