剑指offer 树的子结构 C++

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）
注: 这里需要判断是否是子结构, 不需要判断是子树.
这个和另外一个题很像,Leetcode 572. 另一个树的子树

思路

题目中说了,空树不是任意一个树的子结构，这一点需要注意。
首先，思路是比较简单的，通过一个递归来实现。

1. 首先，如果有任何一棵树是空树，那么直接返回false；
2. 首先判断根节点开始是否是子结构，如果是那么直接返回true；首先判断根节点开始是否是子结构，如果是那么直接返回true；
3. 如果根节点开始不是子结构，那么就判断左子树是否是子结构。类推再判断右子树。

子函数fun() :
用于判断是否是子结构
历遍到root2，如果都相等，那么就找到子结构了。所以用if(!root)来控制。

解答

/*
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 || pRoot2==NULL)
            return false;
        if(fun(pRoot1,pRoot2))
            return true;
        return HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2);
    }
    //子函数fun（）用于判断树root2是否是root1的
    bool fun(TreeNode* root1, TreeNode* root2)
    {
        if(!root2)    //自己在这里纠结了，一开始没有想到这个控制条件。
            return true;
        if(!root1)
            return false;
        if(root1->val != root2->val)
            return false;
        return fun(root1->left,root2->left) && fun(root1->right,root2->right);
    }
};