剑指offer——树的子结构
题目描述:
链接:https://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88?toCommentId=2723570
来源:牛客网
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
题目分析:
//递归
//如果树1包含树2的子树,树1的每个节点都可能与树2的根节点是对应节点。因此需要先看根节点
//如果不是,再看左节点,如果不是,再看节点。
//怎么判断树A的根节点与树B根节点对应时树B是不是树A的子树呢?
//同样递归
//如果B先遍历完,返回true
//如果A先遍历完,返回false
//树B如果对应节点值不等,则一定不是,返回false
//如果相等,那么返回true的充要条件是B的左子树和右子树分别是A的左子树和右子树的子树。
AC代码:
/*
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 result = false;
if (pRoot1 == NULL || pRoot2 == NULL)
return false;
result = Tree1HaveTree2(pRoot1, pRoot2);//先看根节点与树B是对应节点时是否满足
if (!result)//如果不满足
result = HasSubtree(pRoot1->left, pRoot2);//看左孩子点与树B是对应节点时是否满足
if (!result)如果依然满足
result = HasSubtree(pRoot1->right, pRoot2);//看右孩子点与树B是对应节点时是否满足
return result;//返回
}
private:
bool Tree1HaveTree2(TreeNode *pRoot1, TreeNode *pRoot2)
{
if (pRoot2 == NULL)//如果B先遍历完,返回true
return true;
if (pRoot1 == NULL)如果A先遍历完,返回false
return false;//
if (pRoot1->val != pRoot2->val)//如果两节点值不等,返回false
return false;//
return (Tree1HaveTree2(pRoot1->left, pRoot2->left) &&Tree1HaveTree2(pRoot1->right, pRoot2->right)) ;
//如果B的左子树和右子树分别是A的左子树和右子树的子树,返回true;否则返回false
}
};