题目
给你一个整数数组 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的子序列最小的那个。足以是序列,即序列ab,a一定在b前面。可以建立一个栈用来保存目前最优的序列遍历数组如果栈未满(<k)添加,添加时要比较栈顶的元素,在保存后面有足够多的元素可以存入栈的前提下,比要添加的元素大的要出栈。
解题方法
建立栈ans大小为看,遍历,读入元素,当栈顶元素比他大并且栈里有元素并且,n-m-i>k即剩下的元素至少能填满栈的情况下,出栈。这样可以找该元素合适的位置,入栈,注意要判断栈否满了,满了则意味着,这个元素不能入栈
代码
class Solution {
public int[] mostCompetitive(int[] nums, int k) {
int[] ans=new int[k];
int n=nums.length;
int m=0;
for(int i=0;i<n;i++){
int x=nums[i];
while(m>0&&x<ans[m-1]&&n+m-i>k){
m--;
}
if(m<k){
ans[m++]=x;
}
}
return ans;
}
}