剑指offer - 树的子结构

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

代码

解题思路 ，两个递归， 第一个递归遍历是查找val相同的节点 ，第二个递归遍历是是否为子结构
查找到val相同的节点时 ， 调用isSubtree 遍历是否为子树，不是则调用自身。

/*
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)
    {
        if(pRoot1==NULL)
            return false;
        if(pRoot2 == NULL)
            return false;
        return (isSubtree(pRoot1,pRoot2))||HasSubtree(pRoot1->left , pRoot2) || HasSubtree(pRoot1->right , pRoot2);
    }
    private:
    bool isSubtree(TreeNode * r1 , TreeNode * r2)
    {
        if(r2 == NULL)
            return true;
        if(r1 == NULL)
            return false;
        if(r1->val != r2->val){
            return false;
        }
        return isSubtree(r1->left,r2->left) && isSubtree(r1->right,r2->right);
        
    }
    
};