[LEECODE每日一题]找出最具竞争力的子序列

好久没有更新优快云了,这段时间学业压力比较忙所以没有时间写,今天有时间来看看LEECODE的每日一题,碰巧刷到了这样一道题;

题目给的很清楚,既输入一个序列要求给定一个子序列长度,让其输出为一个最有"竞争力"的序列,说白了就是在所有子序列比较中,处于靠前位置的元素要尽可能的小,而且这个子序列是按照顺序进行排列的。

这里我们可以利用单调栈的想发来完成一道题目,具体怎么做呢?

我们可以发现,我们越小的数字放置的位置越前，对应的子序列越具竞争力。

因为子数组是顺序的缘故,所以我们可以拿后面每个比前面小的数字进行逐一比较,如果比前面的一个位置的小,我们就让前面位置的数字进行出栈,再对出栈前一个比较,直到满足比前面的一个栈的元素大为止,如果比前面的数字大我们就直接入栈。然后再拿数组中剩下的元素进行比较。

当然我们如果不停的入栈出栈,我们怎么保证里面的元素是我们想要的子序列长度的大小呢?这里我们可以用这样一种方式进行判断既如果满足

栈里面的元素个数+剩余要比较的数目>最具竞争力的子序列大小

我们可以在进行下一次比较,如果不满足,那么数组剩余元素就直接补充到我们最具竞争力的子序列中

具体代码如下

#include <stdio.h>
#include <stdlib.h>

int* mostCompetitive(int* nums, int numsSize, int k, int* returnSize) {
    int n = numsSize;
    int* res = (int*)malloc(sizeof(int) * n);
    int pos = 0;
    for (int i = 0; i < n; i++)
    {
        while (pos > 0 && n - i + pos > k && res[pos - 1] > nums[i]) 
        {
            pos--;
        }
        res[pos++] = nums[i];
    }
    return res;
}


int main()
{
	int a[] = { 3,2,5,6,9,2,3,1,2,3,3,2 };
	int sz = sizeof(a)/sizeof(int);
    int k = 6;
	int* returnvalue = mostCompetitive(a, sz,k, 5);
    printf("最有竞争力的数组为:");
    int i;
    for (i = 0; i < k; i++)
    {
        printf("%d", *(returnvalue + i));
    }
	return 0;
}