《剑指offer》-树的子结构

题目描述

输入两颗二叉树A，B，判断B是不是A的子结构。

解法：就是先找到相等的根节点，然后再子树遍历。所以要用到两个递归函数，根节点的递归函数，子树的递归函数。

注意子树的递归函数中应该先判断子树是否遍历完为空，则返回true，每次遍历前都应该判断根节点是否存在。这样就完美了。

/*
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* p1, TreeNode* p2) {
        if(!p2) return true;//要先判断子树是否为空。这是为什么呢？如果恰好主树遍历完，子树也遍历完，那么就会出错先
        if(!p1) return false;//需要判定根节点是否为空，这是遍历前的做法
        if(p1->val != p2->val) return false;
        return IsSubtree(p1->left, p2->left) && IsSubtree(p1->right, p2->right); //遍历子树左右子节点

    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {

        bool flag = false;
        if(pRoot1 && pRoot2) {
            if(pRoot1->val == pRoot2->val) {  //寻找到相等的根节点，然后再子树遍历
                flag = IsSubtree(pRoot1, pRoot2);
            }
            if(!flag) flag = HasSubtree(pRoot1->left, pRoot2);  //没有找到相应的根节点，左子树遍历
            if(!flag) flag = HasSubtree(pRoot1->right, pRoot2);  //右子树遍历，找到相等的根节点
        }
        return flag;
    }

};