力扣 1673. 找出最具竞争力的子序列(单调递增栈)

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

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

题解:

读题可知我们很显然可以使用栈来解决此类问题。

我们创建一个栈,然后开始遍历nums数组。由于入栈情况较多,因此我们分析出栈的条件即可,这样的话不出栈即为入栈。
因此我们分析出栈条件:

首先是栈不为空且要满足栈外元素小于栈顶元素。因此这种情况下我们应该让老元素出栈,让新元素入栈。这是因为我们要选择“最具竞争力”的序列,所以应该尽可能的让栈内每个元素都尽可能的小。

但是也不是只有这些条件,因为很可能经过上述更新后的确使栈内元素尽可能的小了,但是可能填不满栈了,即返回的子序列的长度不够了,因此我们还需要判断如若参与更新其长度能否组成要求的k

注意:只要其后面剩余的长度够其满足k即可,也许有人会想,我们不是要维护一个单调递增栈吗,这样怎么能确保我们维护的一定是单调递增栈呢?
这时你需要注意,我们其实并没有要完全的去为维护一个单调递增栈,我们只是尽可能的去维护一个单调递增栈,因为注意这样才能充分发挥每一个元素的价值。但是如果维护不成我们其实也“尽力了”。
并且有人会认为这样的话我们后面因为没有维护,所以可能不是最具竞争力的序列了。但你要发现,在后面“无序”的序列之前,我们都是尽可能的去维护了,并且比较竞争力是从头部开始的,因此只要头部“竞争力”够强,我们不用管后面的子列了。

代码:

int *mostCompetitive(int *nums, int numsSize, int k, int *returnSize)
{
    
    
    int *stack = (int*)malloc(sizeof(int) *numsSize);
    int top = 0;

    for (int i = 0; i < numsSize; i++) 
    {
    
    
        /* 当前栈非空 当前遍历到的数字小于栈顶的数字 并且当前遍历到的数字之后的数字个数满足k个序列的长度 */
        while (top > 0 && stack[top - 1] > nums[i] && numsSize - i + top > k) 
        {
    
    
            top--;
        }
        /* 入栈 */
        stack[top++] = nums[i];
    }
    
    *returnSize = k;
    return stack;
}

猜你喜欢

转载自blog.csdn.net/xiangguang_fight/article/details/115251067
今日推荐