面试题:树的子结构

题目描述

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

分析：

应该有两个递归：

• 递归一：依此检查两棵树的是不是相同（包含）
• 递归二：一次以A树的每个节点为根，与B进行递归一的检查

代码：

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