树的子结构

题目描述

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

思路：

//1.首先需要递归pRoot1树，找到与pRoot2根一样的节点，这需要一个遍历
//2.找到相同的根节点后，要判断是否子树，仍需要一个一个遍历对比
//树的遍历我们一般就用递归来做，那么根据分析需要两个递归函数如下：

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    bool isSubtree(TreeNode* r1, TreeNode* r2){
         //此处为p2 == null 是匹配完成的条件
         //最开始p2肯定不为NULL,这是在主程序HasSubtree中判断过的。
         //递归中，如果p2为空了，则表示上一层的递归中的p2已经匹配完了 
        if(r2 == NULL)  
            return true;
        if(!r1)
            return false;
        if(r1->val == r2->val) //判断两个树的节点是否相等
            return isSubtree(r1->left,r2->left) && isSubtree(r1->right,r2->right);//如果节点相等，判断左右子树是否分别相等
        else
            return false;
    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL || pRoot2 == NULL)
            return false;
        return isSubtree(pRoot1,pRoot2) || HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2);
        //短路特性，判断pRoot2是否是pRoot1的子树
        //判断pRoot2是否是pRoot1左子树的子树
        //判断pRoot2是否是pRoot1右子树的子树
    }
};