【LeetCode 剑指Offer 26】树的子结构 （递归）

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false
示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true
限制：0 <= 节点个数 <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof

分析

判断B是否是A的子树，即判断A中是否有一个子树与B相等，如何判断两棵树相等？

如果要判断A树和B树相等（从A点开始是否与B都有相匹配的点）：
如果B树已空，则成立，因为B的节点已全部匹配，返回；
如果A树已空，则不成立，因为此时B树还有未匹配的点，返回；
若都不空，则A点的值要等于B点的值，并且A点左右子树要等于B点左右子树

思考一下，如果B是A的子树，有这几种情况：从A点开始与B都有相匹配的点，A的左右子树开始与B有相匹配的点，从A点的左右子树的左右子树开始与B有相匹配的点…
用上面的思想去进行递归操作。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isSubStructure(TreeNode* A, TreeNode* B){
        if(A == NULL || B == NULL) return false;
        return solve(A, B) || isSubStructure(A->left, B) || isSubStructure(A->right, B);
    }
    bool solve(TreeNode* A, TreeNode* B){
        if(B == NULL) return true;
        if(A == NULL) return false;
        return A->val == B-> val && solve(A->left, B->left) && solve(A->right, B->right);
    }
};