剑指offer第16题：树的子结构

16、树的子结构

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

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

例如:

给定的树 A:

     3
    / \
   4   5
  / \
 1   2

给定的树 B：

   4 
  /
 1
 

返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 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

思路:
从根前节点出发，每次都比较根节点，如果不一样，那就直接返回false，如果一样，那就继续判断A的左子树是否和我B的左子树一样，并且A的右子树是否和我B的右子树一样
。如果都一样，那就继续（递归）。 判到最后没有被false掉，那就是B已经是空的了。那么B就一定是A的子树。

图解：

代码：

/**
 * 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 || !B) return false;      //约定空树不是任意一个树的子结构
        if(isPart(A, B)) return true;       //如果是的话，这里会直接返回true，否者就是两棵树其中一颗先空
        return isSubStructure(A->left, B) || isSubStructure(A->right, B);
    }

    bool isPart(TreeNode* A, TreeNode* B) {
        if(!B) return true;     //如果B被匹配到空，就说明这课树是A的子树
        if(!A || A->val != B->val) return false;        //B没空，A空了，或者是元素不一样，就不是子树

        return isPart(A->left, B->left) && isPart(A->right, B->right);      //递归判断左子树，右子树
    }
};