【剑指Offer系列】54-二叉搜索树的第K大节点（list.get）

马金朋

已于 2024-06-30 12:15:01 修改

阅读量150

点赞数 1

文章标签：剑指Offer54 二叉搜索树的第K大节点 leetcode23 二叉搜索树中第K小的元素 list.get 中序遍历

于 2020-09-14 22:59:33 首次发布

本文链接：https://blog.csdn.net/tmax52HZ/article/details/108183387

版权

题目：二叉搜索树的第K大节点

举例：输入二叉搜索树，如图示，第三大的节点是6

在这里插入图片描述

思路

中序遍历，然后获取第k个数

public class KthSmallest {
    public static void main(String[] args) {
        TreeNode root = new TreeNode(3);
        TreeNode n1 = new TreeNode(1);
        TreeNode n2 = new TreeNode(4);
        TreeNode n3 = new TreeNode(2);
        root.left =n1;
        root.right = n2;
        n1.right = n3;
    }

    public static int kthSmallest(TreeNode root, int k) {
        List<Integer> list = new ArrayList<>();
        midTraverse(list, root);
        return list.get(k - 1);
    }

    private static void midTraverse(List<Integer> list, TreeNode root) {
        if (root == null) {
            return;
        }
        if (root.left != null) {
            midTraverse(list, root.left);
        }
        list.add(root.val);
        if (root.right != null) {
            midTraverse(list, root.right);
        }
    }
}