概述
最近很忙,都没有时间去弄算法题,自己不在深圳这几天,虽然不是很忙,但都是碎片化的时间,碎片化的时间搞不了力扣,毕竟像我这种小白,难得今天中午没有休息,这样就来打一次LeetCode卡吧,领个10积分虽然不是目标,但是应该值得坚持。
题目
1673. 找出最具竞争力的子序列 - 力扣(LeetCode)
给你一个整数数组 nums
和一个正整数 k
,返回长度为 k
且最具 竞争力 的 nums
子序列。
数组的子序列是从数组中删除一些元素(可能不删除元素)得到的序列。
在子序列 a
和子序列 b
第一个不相同的位置上,如果 a
中的数字小于 b
中对应的数字,那么我们称子序列 a
比子序列 b
(相同长度下)更具 竞争力 。 例如,[1,3,4]
比 [1,3,5]
更具竞争力,在第一个不相同的位置,也就是最后一个位置上, 4
小于 5
。
示例 1:
输入:nums = [3,5,2,6], k = 2 输出:[2,6] 解释:在所有可能的子序列集合 {[3,5], [3,2], [3,6], [5,2], [5,6], [2,6]} 中,[2,6] 最具竞争力。
示例 2:
输入:nums = [2,4,3,3,5,4,9,6], k = 4 输出:[2,3,3,4]
提示:
1 <= nums.length <= 105
0 <= nums[i] <= 109
1 <= k <= nums.length
新知识学习
Deque
我们知道,Queue是队列,只能一头进,另一头出。
如果把条件放松一下,允许两头都进,两头都出,这种队列叫双端队列(Double Ended Queue),学名Deque /dek/。
Java集合提供了接口Deque来实现一个双端队列,它的功能是:
既可以添加到队尾,也可以添加到队首;
既可以从队首获取,又可以从队尾获取。
我们来比较一下Queue和Deque出队和入队的方法:
Queue | Deque | |
添加元素到队尾 | add(E e) / offer(E e) | addLast(E e) / offerLast(E e) |
取队首元素并删除 | E remove() / E poll() | E removeFirst() / E pollFirst() |
取队首元素但不删除 | E element() / E peek() | E getFirst() / E peekFirst() |
添加元素到队首 | 无 | addFirst(E e) / offerFirst(E e) |
取队尾元素并删除 | 无 | E removeLast() / E pollLast() |
取队尾元素但不删除 | 无 | E getLast() / E peekLast() |
对于添加元素到队尾的操作,Queue提供了add()/offer()方法,而Deque提供了addLast()/offerLast()方法。添加元素到队首、取队尾元素的操作在Queue中不存在,在Deque中由addFirst()/removeLast()等方法提供。
解题思路
- 根据题目对竞争力的定义,可以发现越小的数字放置的位置越前,对应的子序列越具竞争力。因此可以用类似单调栈的思想尽量将更小的元素放到子序列的前面。
- 当前元素大于等于栈顶元素,入栈;否则,将栈中大于当前元素的数一直出栈,再将当前元素压栈,以此保证栈中最小字典序的顺序。
- 其中记录删除了多少元素,题目保留k个元素,即删除n-k个元素,若删除的数目达到上限,则保留栈中元素不再删除。
- 遍历一遍后,若栈中元素数目大于k个(即n-k>0),将多余元素出栈。
代码
class Solution {
public int[] mostCompetitive(int[] nums, int k) {
int len = nums.length;
if (k >= len) {
return nums;
}
int removeCount = len - k;
//双端队列
Deque<Integer> deque = new ArrayDeque<>();
for (int num : nums) {
while (removeCount > 0 && !deque.isEmpty() && deque.peekLast() > num) {
deque.removeLast();
removeCount--;
}
deque.addLast(num);
}
//还有剩余的位置没有移除,从最后移除
for (int i = 0; i < removeCount; i++) {
deque.removeLast();
}
int[] res = new int[k];
int idx = k - 1;
for (int i = 0; i < k; i++) {
res[idx] = deque.removeLast();
idx--;
}
return res;
}
}
最后还是强调,学习算法要学习数据结构,今天花费了点时间学习Deque双端对象,让整个解题思路得到了更简单和高效。