LC每日一题｜20240524 - 1673. 找出最具竞争力的子序列

LC每日一题｜20240524 - 1673. 找出最具竞争力的子序列

给你一个整数数组 nums 和一个正整数 k ，返回长度为 k 且最具 竞争力 的 **nums 子序列。

数组的子序列是从数组中删除一些元素（可能不删除元素）得到的序列。

在子序列 a 和子序列 b 第一个不相同的位置上，如果 a 中的数字小于 b 中对应的数字，那么我们称子序列 a 比子序列 b（相同长度下）更具 竞争力 。 例如，[1,3,4] 比 [1,3,5] 更具竞争力，在第一个不相同的位置，也就是最后一个位置上， 4 小于 5 。

提示：

• 1 <= nums.length <= 10^5
• 0 <= nums[i] <= 10^9
• 1 <= k <= nums.length

题目等级：Medium

解题思路

很简单的一道单调栈应用题～

维护一个单调栈，在遍历nums的过程中，同时满足以下情况时弹出栈顶元素

• 单调栈不为空
• 单调栈栈顶元素比当前元素大
• 未入栈元素数量加当前栈内元素数量需大于k

然后将栈内元素打入数组内返回即可，当然也可以直接开一个数组模拟栈操作，还能省下一些空间～

AC代码

class Solution {
    fun mostCompetitive(nums: IntArray, k: Int): IntArray {
        val stack = Stack<Int>()
        for (i in nums.indices) {
            val n = nums[i]
            while (stack.isNotEmpty() && stack.peek() > n && k - stack.size < nums.size - i) {
                stack.pop()
            }
            stack.push(n)
        }
        while (stack.size > k) stack.pop()
        val res = IntArray(k)
        for (i in res.indices) {
            val j = res.size - i - 1
            res[j] = stack.pop()
        }
        return res
    }
}

时间复杂度：O(n)

空间复杂度：O(n)