提交网址: http://www.nowcoder.com/practice/6e196c44c7004d15b1610b9afca8bd88?tpId=13&tqId=11170
或 http://www.lintcode.com/zh-cn/problem/subtree/ (难度: Easy)
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
要判断二叉树 A 中是否存在和树 B 结构一样的子结构,可以分成两步: 1.在树 A 中找到和 B 的根结点的值一样的结点R; 2.再判断树 A 中以 R 为根结点的子树是不是包含和树 B 一样的结构。
AC代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
bool Tree1HaveTree2(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot2 == NULL) return true;
if(pRoot1 == NULL) return false; // 前两个递归出口条件的顺序有影响
if(pRoot1->val != pRoot2->val) return false;
return (Tree1HaveTree2(pRoot1->left, pRoot2->left))&&(Tree1HaveTree2(pRoot1->right, pRoot2->right));
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool res = false;
if(pRoot1 != NULL && pRoot2 != NULL)
{
if(pRoot1->val == pRoot2->val) res = Tree1HaveTree2(pRoot1, pRoot2);
if(res == false) res = HasSubtree(pRoot1->left, pRoot2);
if(res == false) res = HasSubtree(pRoot1->right, pRoot2);
}
return res;
}
};