剑指offer（第二版）——二叉搜索树的第k大节点

最新推荐文章于 2024-09-18 20:33:32 发布

且思且行且珍惜且成长

最新推荐文章于 2024-09-18 20:33:32 发布

阅读量160

点赞数

分类专栏：剑指offer（第二版）Java详解文章标签：算法面试数据结构

本文链接：https://blog.csdn.net/qq_41866626/article/details/120562754

版权

剑指offer（第二版）Java详解专栏收录该内容

67 篇文章 2 订阅

订阅专栏

PS：《剑指offer》是很多同学找工作都会参考的一本面试指南，同时也是一本算法指南（为什么它这么受欢迎，主要应该是其提供了一个循序渐进的优化解法，这点我觉得十分友好）。现在很多互联网的算法面试题基本上可以在这里找到影子，为了以后方便参考与回顾，现将书中例题用Java实现（第二版），欢迎各位同学一起交流进步。

GitHub： https://github.com/Uplpw/SwordOffer。

剑指offer完整题目链接： https://blog.csdn.net/qq_41866626/article/details/120415258

1 题目描述

找出二叉搜索树的第k大节点。例如，在下图的树里，第3大节点的值为4，输入该树的根节点，3，则输出4。

        5
       / \
      3   7
    / \   / \
   2  4  6   8

leetcode链接： 二叉搜索树的第k大节点（以下代码已测试，提交通过）

2 测试用例

一般是考虑功能用例，特殊（边缘）用例或者是反例，无效测试用例这三种情况。甚至可以从测试用例寻找一些规律解决问题，同时也可以让我们的程序更加完整鲁棒。

（1）功能用例：完全与非完全二叉树。

（2）边缘用例：只有一个节点，只有左节点，只有右节点。

（3）无效用例：树为空。

3 思路

分析：

由于是二叉搜索树，索引其中序遍历是顺序的，下面是两种解决方式。

（1）利用二叉搜索树的中序遍历，将其存放到list查找，比较简单直观

（2）利用二叉搜索树的中序遍历的倒序（左中右----右中左，这样访问的顺序是倒序的），通过全局变量控制访问的次数，直到达到第k大元素。这样就可以减少list的空间。

4 代码

树的结构：

public class TreeNode {
    public int val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode(int val) {
        this.val = val;
        this.left = null;
        this.right = null;
    }
}

算法实现：

import java.util.ArrayList;
import java.util.List;

public class KthNodeInBST {
	// 1. 使用list存储中序遍历的结果。
    public static int kthLargest(TreeNode root, int k) {
        if (root == null || k < 0) {
            return -1;
        }
        List<TreeNode> list = new ArrayList<>();
        traverse(root, list);
        int length = list.size();
        if (k > length) {
            return -1;
        }
        return list.get(length - k).val;
    }

    public static void traverse(TreeNode node, List<TreeNode> list) {
        if (node == null) {
            return;
        }
        traverse(node.left, list);
        list.add(node);
        traverse(node.right, list);
    }

    int count = 0;
    int res = -1;

    // 2. 优化不使用额外空间，递归遍历（采用右-根-左倒序）第k大，则每次减一，直到为0
    public int kthLargest2(TreeNode root, int k) {
        if (root == null || k < 0) {
            return -1;
        }
        traverse2(root, k);
        return res;
    }

    public void traverse2(TreeNode node, int k) {
        if (node == null) {
            return;
        }
        traverse2(node.right, k);
        count++;
        if (k == count) {
            res = node.val;
            return;
        }
        traverse2(node.left, k);
    }

    public static void main(String[] args) {
        // 2 3 4 5 6 7 8
        TreeNode root = new TreeNode(5);
        root.left = new TreeNode(3);
        root.left.left = new TreeNode(2);
        root.left.right = new TreeNode(4);
        root.right = new TreeNode(7);
        root.right.left = new TreeNode(6);
        root.right.right = new TreeNode(8);

//        System.out.println(kthLargest(root, 3)); //6
//        System.out.println(kthLargest(root, 6)); //3
//        System.out.println(kthLargest(root, 8)); //null

        System.out.println("count:" + new KthNodeInBST().kthLargest2(root, 3)); // 6
        System.out.println(new KthNodeInBST().kthLargest2(root, 6)); // 3
        System.out.println(new KthNodeInBST().kthLargest2(root, 8)); // null
    }
}