给你一个整数数组 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 <= 10**5
0 <= nums[i] <= 10**9
1 <= k <= nums.length
题解及其思路在代码上面的注释里。
class Solution:
def mostCompetitive(self, nums: List[int], k: int) -> List[int]:
#1.栈的数据结构。
#2.先凑k,如果遇到比栈顶等元素小的元素可以进行不断跟换。
#3.不断更换的条件:栈不为空,遇到的元素比栈顶的小,栈的长度加上nums长度减去已经遍历的长度要大于k:
#4.如果栈的长度小于k,就加入元素不用论是否比栈顶的大小
stack = []
for i,x in enumerate(nums):
while stack and x < stack[-1] and len(stack) + len(nums) - i > k:
stack.pop()
if len(stack) < k:
stack.append(x)
return stack
关于什么是栈(一种数据结构):
关于为什么这个思路是对的,可以举些例子来证明(自己想一想更能想明白逻辑正确的问题)