找出最具竞争力的子序列

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


来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-the-most-competitive-subsequence
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

维护一个单调栈（如果当前的元素小于栈顶元素，则将栈中大于当前元素的所有元素出栈，将当前元素入栈）

这里需要对单调栈进行一下修改，因为最多只能删除n-k个元素，要剩下至少k个元素，如果当前队列中的元素已经删除了n-k个，那么往后就不能出栈，只能入栈元素。

class Solution {
    public int[] mostCompetitive(int[] nums, int k) {
        if (nums == null || nums.length == 0) return new int[]{};
        Deque<Integer> stack = new LinkedList<>();

        int n = nums.length;

        int maxDelete = n - k;
        stack.push(nums[0]);
        for (int i = 1; i < nums.length; i++) {
            while (maxDelete > 0 && !stack.isEmpty() && stack.peek() > nums[i]) {
                stack.pop();
                maxDelete--;
            }
            stack.push(nums[i]);
        }

        int index = k - 1;
        int[] res = new int[k];

        int size = stack.size();
        for (int i = 0; i < size - k; i++) stack.pop();
        while (!stack.isEmpty()) {
            res[index--] = stack.pop();
        }
        return res;
    }
}

赏

谢谢你请我吃糖果

微信