剑指offer 树的子结构（C++）

题目描述

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

解题思路

这种问题一定要用递归的方法来解答。

先写一个bool类型的匹配函数，从A树的头结点开始与B树进行匹配，如果匹配成功，返回true，否则从A->left开始与B树进行结点匹配，调用匹配函数，A的所有位置均与B树进行匹配，直到匹配成功为止，均不成功，返回false。

代码实现

/*
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* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot2 == NULL) // 这条语句必须在前面 
            return true;
        if(pRoot1 == NULL) 
            return false;
        if(pRoot1->val == pRoot2->val)
            if(IsSubtree(pRoot1->left, pRoot2->left) && IsSubtree(pRoot1->right, pRoot2->right))
                return true;
		//return IsSubtree(pRoot1->left, pRoot2->left) && IsSubtree(pRoot1->right, pRoot2->right) // 与上式等同
        return false;
    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL || pRoot2 == NULL)
            return false;
        if(IsSubtree(pRoot1, pRoot2) || HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2))
            return true;
        return false;
         /*
        return IsSubtree(pRoot1, pRoot2) || 
            HasSubtree(pRoot1->left, pRoot2) || 
            HasSubtree(pRoot1->right, pRoot2);*/ // 与上式等同
    }
};

注释

 if(pRoot2 == NULL) // 这条语句必须在前面 
	return true;
if(pRoot1 == NULL) 
	return false;

在匹配函数中，必须先对B树进行判空，毕竟B树是A的子树，如果在递归的过程中，A的某个结点的孩子结点为空并且B的某个孩子结点也为空，这种情况是可以存在两者匹配的情况的，但是如果将对A判空的代码写在前面，直接返回false的话，则会对问题进行误判；B树的某一个分支为空无论A树该分支为空与否，B树都是A树的子树；但是如果B树分支不空，A树的分支却为空，则B树一定不是A树的子树。以上就表明对B树的判空一定要写在对A树的判空的前面的原因。

树的递归问题，一直是很困扰的编程问题，递归函数的特点也是难想易写。