剑指offer(26)

剑指offer(26)

剑指 Offer 26. 树的子结构

输入两棵二叉树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

思路：

利用两个递归来进行判断。第一个递归是判断B是不是A的子结构

第二个递归用来判断A为根节点的子树和B是否结构相等

class Solution {
public:
    //函数定义B是不是A的子结构
    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if(B==NULL||A==NULL)return false;//判断B是不是A的子结构 所以AB任意一个为空都不行
        if(A->val==B->val){//如果A点值等于B点值则可以继续判断
            if(traver(A,B))return true;
        }
        //没找到就去继续向下寻找
        return isSubStructure(A->left,B)||isSubStructure(A->right,B);
    }
    //判断A节点结构是否和B相等
    bool traver(TreeNode* A,TreeNode* B){
        //如果A已经为空，但是B不是空，就说明AB不一样
        if(A==NULL&&B!=NULL)return false;
        //如果B判断完了 就说明AB一样
        if(B==NULL)return true;
        //如果每个结点值不同 说明AB不一样
        if(A->val!=B->val)return false;
        //去向下遍历
        return traver(A->left,B->left)&&traver(A->right,B->right);
    }
};