leetcode每日一题系列-最小K个数-「最小优先队列」

336 阅读1分钟

leetcode-面试题 17.14- 最小K个数

[博客链接]

菜🐔的学习之路

掘金首页

[题目描述]

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

 

示例:

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

[题目链接]

leetcode题目链接

[github地址]

代码链接

[思路介绍]

思路一:优先队列

  • 这题中等属实是没啥必要
  • 排序/堆都可以直接做
 public int[] smallestK(int[] arr, int k) {
            int[] res = new int[k];
            PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
            for (int num: arr
                 ) {
                priorityQueue.add(num);
            }
            for (int i = 0; i < k && !priorityQueue.isEmpty(); i++) {
                res[i] = priorityQueue.poll();
            }
            return res;
        }

  • 时间复杂度O(nlgn)
  • 空间复杂度O(n)