题目地址:
https://leetcode.com/problems/find-the-most-competitive-subsequence/
给定一个数组 A A A,求其字典序最小的长 k k k的子序列。
思路是单调栈。求长 k k k的子序列相当于在 A A A里删掉 l A − k l_A-k lA−k个数字。我们用一个变量 r r r标记还可以删多少个数字,一开始赋值为 l A − k l_A-k lA−k。接着开一个从栈底到栈顶严格单调上升的栈,如果栈非空并且栈顶大于新来的数,并且还能删数字(指 r > 0 r>0 r>0成立),则出栈,并将 r r r减 1 1 1,直到栈空或者栈顶小于等于新来的数或者 r = 0 r=0 r=0(这时没法删了),接着将新来的数入栈(如果栈满,说明新来的数无法入栈,则多删一个数,将 r r r减去 1 1 1)。最后栈底到栈顶的数组即为所求。算法正确性证明,只需要注意每次可以删的时候(这里的删主要指的是pop栈顶),删总是比不删好。代码如下:
class Solution {
public:
vector<int> mostCompetitive(vector<int>& a, int k) {
vector<int> res(k);
int top = 0, n = a.size();
k = n - k;
for (int i = 0; i < n; i++) {
while (top && res[top - 1] > a[i] && k) {
top--;
k--;
}
if (top < res.size()) res[top++] = a[i];
else k--;
}
return res;
}
};
时间复杂度 O ( l A ) O(l_A) O(lA),空间 O ( 1 ) O(1) O(1)。