聊聊判断二叉树A中是否包含子树B

开发 前端
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)B是A的子结构, 即 A中有出现和B相同的结构和节点值。

[[437107]]

Leetcode : https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof

“GitHub : https://github.com/nateshao/leetcode/blob/main/algo-notes/src/main/java/com/nateshao/sword_offer/topic_20_isSubStructure/Solution.java

判断二叉树A中是否包含子树B

“题目描述 :输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)B是A的子结构, 即 A中有出现和B相同的结构和节点值。例如:给定的树 A:

  1.        3 
  2.       / \ 
  3.     4   5 
  4.   / \ 
  5. 1   2 

给定的树 B:

  1.  
  2.  

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

  1. 输入:A = [1,2,3], B = [3,1] 
  2.  
  3. 输出:false 

示例 2:

  1. 输入:A = [3,4,5,1,2], B = [4,1] 
  2.  
  3. 输出:true 

限制:0 <= 节点个数 <= 10000

解题思路:若树B是树A的子结构,则子结构的根节点可能为树A的任意一个节点。因此,判断树B是否是树A的子结构,需完成以下两步工作:

先序遍历树A中的每个节点nA ; (对应函数 isSubStructure(A, B) )

判断树A中以nA为根节点的子树否包含树B。(对应函数recur(A,B))

算法流程:

“名词规定:树 A 的根节点记作 节点 A ,树 B 的根节点称为 节点 B 。

recur(A, B) 函数:

1.终止条件:

  • 当节点 B为空:说明树 B 已匹配完成(越过叶子节点),因此返回 true ;
  • 当节点 A为空:说明已经越过树 A 叶子节点,即匹配失败,返回 false ;
  • 当节点 A 和 B 的值不同:说明匹配失败,返回 false ;

2.返回值:

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

isSubStructure(A, B)函数:

  • 特例处理 :当树A为空或树B为空时,直接返回false;
  • 返回值 :若树B是树A的子结构,则必满足以下三种情况之一,因此用或|连接;
    • 以节点A为根节点的子树包含树B,对应recur(A,B);
    • 树B是树A左子树的子结构,对应isSubStructure(A. left, B) ;
    • 树B是树A右子树的子结构,对应isSubStructure(A. right, B) ;

以让 2. 3.实质上是在对树A做先序遍历。

复杂度分析:

  • 时间复杂度O(MN): 中M,N分别为树A和树B的节点数量;先序遍历树A占用0(M),每次调用recur(A, B)判断占用O(N)。
  • 空间复杂度O(M):
    • 当树A和树B都退化为链表时,递归调用深度最大。
    • 当M≤N时,遍历树A与递归判断的总递归深度为M ;
    • 当M> N时,最差情况为遍历至树A叶子节点,此时总递归深度为M。
  1. package com.nateshao.sword_offer.topic_20_isSubStructure; 
  2.  
  3. /** 
  4.  * @date Created by 邵桐杰 on 2021/11/23 19:19 
  5.  * @微信公众号 程序员千羽 
  6.  * @个人网站 www.nateshao.cn 
  7.  * @博客 https://nateshao.gitee.io 
  8.  * @GitHub https://github.com/nateshao 
  9.  * @Gitee https://gitee.com/nateshao 
  10.  * Description: 判断二叉树A中是否包含子树B 
  11.  */ 
  12. class Solution { 
  13.  
  14.     /** 
  15.      * 精选解答 
  16.      * @param A 
  17.      * @param B 
  18.      * @return 
  19.      */ 
  20.     public static boolean isSubStructure1(TreeNode A, TreeNode B) { 
  21.         return (A != null && B != null) && (recur1(A, B) || isSubStructure1(A.left, B) || isSubStructure1(A.right, B)); 
  22.     } 
  23.  
  24.     public static boolean recur1(TreeNode A, TreeNode B) { 
  25.         if (B == nullreturn true
  26.         if (A == null || A.val != B.val) return false
  27.         return recur1(A.left, B.left) && recur1(A.right, B.right); 
  28.     } 
  29.      
  30.     /*********************************** 法二 *********************************************/ 
  31.     //遍历A的每一个节点 
  32.     public boolean isSubStructure(TreeNode A, TreeNode B) { 
  33.         if (A == null || B == nullreturn false;//约定空树不是任意一个树的子结构 
  34.         return recur(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B); 
  35.     } 
  36.  
  37.     //同时遍历A和B的相同位置节点 
  38.     boolean recur(TreeNode A, TreeNode B) { 
  39.         //当B某个节点为null,则无需比较了,直接返回true,比较其他节点 
  40.         if (B == nullreturn true
  41.         //如果相同位置的两个节点不相同,则返回false,不用再继续比较了 
  42.         if (A == null || A.val != B.val) return false
  43.         //继续往下遍历比较 
  44.         return recur(A.left, B.left) && recur(A.right, B.right); 
  45.     } 
  46.  
  47.     public class TreeNode { 
  48.         int val; 
  49.         TreeNode left
  50.         TreeNode right
  51.  
  52.         TreeNode(int x) { 
  53.             val = x; 
  54.         } 
  55.     } 

 

参考链接:https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof/solution/mian-shi-ti-26-shu-de-zi-jie-gou-xian-xu-bian-li-p

 

责任编辑:武晓燕 来源: 程序员千羽
相关推荐

2021-10-12 09:25:11

二叉树树形结构

2020-04-27 07:05:58

二叉树左子树右子树

2021-12-03 09:16:03

二叉树打印平衡

2021-04-19 07:47:42

数据结构二叉树Tree

2021-04-20 08:37:14

数据结构二叉树

2013-07-15 16:35:55

二叉树迭代器

2021-03-17 08:19:22

二叉树LeetCode

2023-02-01 07:27:46

序列化二叉树根节点

2021-09-02 11:31:28

二叉搜索树迭代法公共祖先

2021-09-29 10:19:00

算法平衡二叉树

2020-11-25 08:25:02

二叉树节点

2021-05-06 17:46:30

二叉树数据结构

2022-10-26 23:58:02

二叉树数组算法

2020-09-23 18:25:40

算法二叉树多叉树

2021-04-28 20:12:27

数据结构创建

2021-08-26 11:31:11

二叉树数据结构算法

2020-12-22 08:56:51

JavaScript数据结构前端

2018-03-15 08:31:57

二叉树存储结构

2021-08-27 11:36:44

二叉树回溯节点

2023-05-08 15:57:16

二叉树数据结构
点赞
收藏

51CTO技术栈公众号