230. 二叉搜索树中第K小的元素

overdosed_

于 2023-03-31 00:18:34 发布

阅读量51

点赞数

分类专栏： leetcode 文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/weixin_45904682/article/details/129869918

版权

leetcode 专栏收录该内容

82 篇文章 0 订阅

订阅专栏

暴力解法:

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()){
            TreeNode temp = queue.poll();
            list.add(temp.val);
            if(temp.left != null){
                queue.offer(temp.left);
            }
            if(temp.right != null){
                queue.offer(temp.right);
            }
        }
        Integer[] arr = list.toArray(new Integer[list.size()]);
        Arrays.sort(arr);
        return arr[k - 1];
    }
}

利用二叉搜索树 + 中序遍历性质

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        Deque<TreeNode> stack = new ArrayDeque<>();
        while (root != null || !stack.isEmpty()){
            while (root !=null){
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            k--;
            if(k == 0){
                return root.val;
            }
            root = root.right;
        }
        return -1;
    }
}