树的子结构(递归!)

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

题目解析
题目要求查找树 A 中是否存在和树 B 结构一样的子树,只需完成以下两个工作即可:

在树 A 中找到和树 B 的根节点的值一样的节点 root
判断树 A 中以 root 为根节点的子树是不是包含和树 B 一样的结构
算法流程

工作一: 对二叉树进行遍历找到树 A 的根节点和树 B 的根节点相等的位置 root
工作二: 判断树 A 中以 root 为根节点的子树和树 B 具有相同的结构

处理 recur(A, B) 函数

- 终止条件

当节点 B 为空:说明树 B 已完成匹配,返回 true
当节点 A 为空:说明树 A 全部遍历完也无法匹配树 B,返回 false
当节点 A 和 B 的值不等:匹配失败,返回 false
- 返回值

判断 A 和 B 的左子节点是否相等,即 recur(A->left, B->left);
判断 A 和 B 的右子节点是否相等,即 recur(A->right, B->right);

重点思想:将问题拆解

1.找到两个二叉树相等的节点

2.再匹配下面的结构是否相等

在isSubStructure中,通过递归遍历A的每一个节点是否和B的第一个节点相等,如果都不相等,则最终返回false

如果A有一个节点和B相等,则在recur中一起对A和B的左右分支进行比较,中途有一个不相等,则B就不是A的子结构,如果遍历到B的末端都没出现不相等的,则B就是A 的子结构,返回true

class Solution {
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
     // 特判:树 A 为空 或 树 B 为空 时,直接返回 false
     if (A == nullptr || B == nullptr) return false;
    /* 
    1. 判断 B 是 A 的子树,recur(A, B);
    2. 判断 B 是 A 的左子树,isSubisSubStructure(A->left, B);
    3. 判断 B 是 A 的右子树,isSubisSubStructure(A->right, B);
    */
    return recur(A, B) || isSubStructure(A->left, B) || isSubStructure(A->right, B);
    }
    bool recur(TreeNode* A, TreeNode* B) {
        if (B == nullptr) return true;
        if (A == nullptr || A->val != B->val) return false;
        return recur(A->left, B->left) && recur(A->right, B->right);
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值