其他分享
首页 > 其他分享> > 剑指 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

方法:要判断B是否是A的子结构,只需要从根节点开始判断,通过递归的方式比较他的每一个子节点即可。但实际上B如果是A的子结构的话,不一定是从根节点开始的,可能使从中间的某一部分,如果题中的两个树AB的情况。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
    	if (A == null || B == null)
        	return false;
    //先从根节点判断B是不是A的子结构,如果不是在分别从左右两个子树判断,
    //只要有一个为true,就说明B是A的子结构
    	return isSub(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }

    public boolean isSub(TreeNode A, TreeNode B) {
        //这里如果B为空,说明B已经访问完了,确定是A的子结构
        if (B == null)
            return true;
        //如果B不为空A为空,或者这两个节点值不同,说明B树不是
        //A的子结构,直接返回false
        if (A == null || A.val != B.val)
            return false;
        //当前节点比较完之后还要继续判断左右子节点
        return isSub(A.left, B.left) && isSub(A.right, B.right);
    }
}

就是递归的思想,isSub是递归函数,这个是递归函数,isSubStructure是主函数,所以主函数return的时候第一个是isSub,后两个是可能在子树中的情况,所以用isSubStructure。

标签:26,return,Offer,子结构,TreeNode,isSub,节点,isSubStructure
来源: https://blog.csdn.net/weixin_45360952/article/details/120394510