二叉树递归套路(2):判断二叉树是否是搜索二叉树、二叉树的最大距离

本篇继续来聊聊二叉树的递归套路。

一、判断二叉树是否是搜索二叉树

搜索二叉树定义:二叉树中的任意一个以X为头的子树,左子树都比X小,右子树都比X大。(经典的搜索二叉树是没有重复值的)

1、经典做法

中序遍历,结果是递增的,说明这是搜索二叉树。

2、递归套路思路

分析任意一个以X为头的子树,满足以X为头的子树是搜索二叉树的条件(列出所有可能性

1)左子树是搜索二叉树

2)右子树是搜索二叉树

3)左子树的最大值 小于 X

4)右子树的最小值 大于 X

满足这四个条件才能说以X为头的子树是平衡二叉树。

问题来了:此时发现我需要从左子树和右子树中获取的结果不一致(一个是要最小值、一个是要最大值),不好拼凑出递归。咋办?好说,我们直接求全集(小孩子才做选择,成年人不做选择,都要),左子树我们求最大值和最小值,右子树也是。

也就是每次从左子树和右子树中我们都需要 是否搜索二叉树、最大值、最小值 三个数据,尽管我们最后只返回是否搜索二叉树,但是我们需要最大值和最小值来辅助我们判断是否搜索,所以可以定义如下的Info类

/**
 * @author Java和算法学习:周一
 */
public static class Info {
    public boolean isSearch;
    public int max;
    public int min;

    public Info(boolean isSearch, int max, int min) {
        this.isSearch = isSearch;
        this.max = max;
        this.min = min;
    }
}

复制代码

3、递归套路代码

(1)首先判断为空时好不好设置,此时不好设置,节点为空时max和min不好指定,所以节点为空时直接返回null,后面递归时再处理这个null即可。

(2)然后根据列出的所有可能性,编写递归套路的代码,因为要整个形成递归,所以每一步都要返回Info类。(无脑拿到左右子树的Info、拼凑自己的Info、返回自己的Info)

/**
 * 判断是否是搜索二叉树的递归套路写法
 *
 * @author Java和算法学习:周一
 */
public static Info process(Node x) {
    // 为空时,max和min无法赋值,所以此处返回null
    if (x == null) {
        return null;
    }

    // 获取左右子树的数据
    Info leftInfo = process(x.left);
    Info rightInfo = process(x.right);

    // 拼凑自己的数据
    boolean isSearch = true;
    // 当前节点不是搜索二叉树的情况
    // 1.左子树不是搜索二叉树
    // 2.右子树不是搜索二叉树
    // 3.左子树最大值大于当前节点
    // 4.右子树最小值小于当前节点
    if (leftInfo != null && (!leftInfo.isSearch || leftInfo.max >= x.value)) {
        isSearch = false;
    }
    if (rightInfo != null && (!rightInfo.isSearch || rightInfo.min <= x.value)) {
        isSearch = false;
    }

    int max = x.value;
    int min = x.value;
    if (leftInfo != null) {
        max = Math.max(leftInfo.max, max);
        min = Math.min(leftInfo.min, min);
    }
    if (rightInfo != null) {
        max = Math.max(rightInfo.max, max);
        min = Math.min(rightInfo.min, min);
    }

    return new Info(isSearch, max, min);
}

复制代码

(3)主函数调用递归方法获取结果

/**
 * @author Java和算法学习:周一
 */
public static boolean isSearchBinaryTree(Node head) {
    if (head == null) {
        return true;
    }
    return process(head).isSearch;
}

复制代码

所有代码地址:

github.com/monday-pro/…

二、二叉树的最大距离

给定一个二叉树,任何两个节点之间都存在距离,返回整个二叉树的最大距离。两个节点的距离表示从一个节点到另一个节点最短路径上的节点数(包含两个节点自己)

1、递归套路思路

首先分析以X为头的二叉树,它的最大距离的可能性有两类,不经过X、经过X。

111.png

不经过X:需要获取左子树和右子树各自的最大距离

经过X:(左树高度 + 右树高度 + 1)即是最大距离

也就是每次从左子树和右子树中我们都需要 最大距离、高度 两个数据,最后再取 左子树最大距离、右子树最大距离、(左树高度 + 右树高度 + 1) 的最大值即是整个树的最大距离。尽管我们最后只返回最大距离,但是我们需要高度来辅助我们计算最大距离,所以可以定义如下的Info类

/**
 * @author Java和算法学习:周一
 */
public static class Info {
    public int height;
    public int maxDistance;

    public Info(int height, int maxDistance) {
        this.height = height;
        this.maxDistance = maxDistance;
    }
}

复制代码

2、递归套路代码

(1)首先判断为空时好不好设置,此时是好设置的,节点为空时new Info(0, 0),即认为空节点高度为0、最大距离为0。

(2)然后根据列出的所有可能性,编写递归套路的代码,因为要整个形成递归,所以每一步都要返回Info类。(无脑拿到左右子树的Info、拼凑自己的Info、返回自己的Info

/**
 * 求二叉树的最大距离的递归套路写法
 * 
 * @author Java和算法学习:周一
 */
public static Info process(Node x) {
    if (x == null) {
        return new Info(00);
    }

    // 获取左右子树的信息
    Info leftInfo = process(x.left);
    Info rightInfo = process(x.right);

    // 拼凑处理自己的信息
    // 最大高度是 左右子树最大高度 + 1
    int height = Math.max(leftInfo.height, rightInfo.height) + 1;
    // 最大距离是 左子树最大距离、右子树最大距离、(左树高度 + 右树高度 + 1) 三个的最大值
    int maxDistance = Math.max((leftInfo.height + rightInfo.height + 1),
            Math.max(leftInfo.maxDistance, rightInfo.maxDistance));

    return new Info(height, maxDistance);
}

复制代码

(3)主函数调用递归方法获取结果

/**
 * @author Java和算法学习:周一
 */
public static int maxDistance(Node head) {
    if (head == null) {
        return 0;
    }
    return process(head).maxDistance;
}

复制代码

所有代码地址:

github.com/monday-pro/…

这是二叉树递归套路解法的第二篇,结合上一篇是不是发现又有了新的收获呢。这次我们又学会了左右子树需要获取的值不同时如何处理、在节点为空时不好设置值时如何处理等

猜你喜欢

转载自juejin.im/post/7047114581579153444