5614.找出最具竞争力的子序列

题目

给你一个整数数组 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]

方法（单调栈）

单调栈可以用于寻找在一个数组中，一个元素右边比它大的最小元素是谁，以及左边比它小的最大元素是谁。

而反观这道题，要寻找最具竞争力的子序列，不就是不断地在一个元素右边尽可能找到一个比它大的更小的元素吗

算法步骤如下：
准备一个栈，这个栈需要保持从栈底到栈顶元素大小从小到大，依次遍历数组中的每个元素

• 如果该元素小于栈顶，并且这时弹出栈顶不会造成栈中元素和数组中剩余元素加在一起都凑不到k的情况，那么就放心弹出栈顶
• 如果该元素大于栈顶，并且栈中元素尚未到K，那么就将该元素入栈。
• 最后，将这个大小为K的栈弹出作为结果。

代码

public int[] mostCompetitive(int[] nums, int k) {
    Stack<Integer> stack = new Stack<>();
    stack.push(nums[0]);
    for(int i = 1; i < nums.length; i++){
        while(!stack.isEmpty() && stack.peek() > nums[i] && stack.size() + nums.length - i > k){
            stack.pop();
        }
        if(stack.size() < k)
            stack.push(nums[i]);
    }
    int[] res = new int[k];
    for(int i = k - 1; i >= 0; i--)
        res[i] = stack.pop();
    return res;
}