leetcode 230. Kth Smallest Element in a BST

最新推荐文章于 2023-06-03 15:29:39 发布

a1025461748

最新推荐文章于 2023-06-03 15:29:39 发布

阅读量131

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode练习

本文链接：https://blog.youkuaiyun.com/a1025461748/article/details/73180778

Leetcode练习专栏收录该内容

294 篇文章

订阅专栏

本文介绍了一种寻找二叉搜索树中第K小元素的方法，通过中序遍历将节点值存储到列表中，进而快速找到目标元素。同时讨论了在频繁增删操作下如何优化该过程。

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note:
You may assume k is always valid, 1 ? k ? BST's total elements.

Follow up:
What if the BST is modified (insert/delete operations) often and you need to find the kth smallest frequently? How would you optimize the kthSmallest routine?

中序遍历把数存到一个list中，多次读取挺方便的，修改的话直接在list也能操作

public class Solution {
    List<Integer> list;
    public int kthSmallest(TreeNode root, int k) {
        list = new ArrayList<>();
        helper(root);
        return list.get(k-1);
    }
    
    private void helper(TreeNode root){
        if(root==null) return;
        helper(root.left);
        list.add(root.val);
        helper(root.right);
    }
}

不过大概题目不是这个意思吧。。。那还是在遍历的时候统计一下数目好了