树的子结构

题目描述

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

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    //注意，这里的isSame指的是root2是root1的子结构，只要root1的前面部分和root2匹配就可以了，即root1可能比root2要大一些，
    bool isSame(TreeNode* root1,TreeNode* root2)
    {
        if(root2==NULL)
            return true;
        else if(root1==NULL)
            return false;
        //到了这里说明已经两个节点都不会空了
        if(root1->val!=root2->val)
            return false;
        
        return isSame(root1->left,root2->left) && isSame(root1->right,root2->right);
    }
    
    
    bool HasSubtree(TreeNode* A, TreeNode* B)
    {
        //只要当两个树都不为空的时候，才有可能返回true。否则返回false
        if(A&&B)
        {
            if(A->val == B->val && isSame(A,B))
                return true;
            else
                return HasSubtree(A->left,B) || HasSubtree(A->right,B);
        }
        
        return false;
    }
};