这里将4个类似的题进行汇总,都是通过删除k个元素/重复元素,使得剩下的数组最大/最小,都是采用类单调栈的方法。
单调栈的思路,但是由于每个元素至少一个或者删除个数的限制,栈其实并不是完全单调的。
思路就是 遇到一个新字符 如果比栈顶小 并且在新字符后面还有和栈顶一样的 就把栈顶的字符抛弃了
class Solution: def removeDuplicateLetters(self, s) -> int: stack = [] remain_counter = collections.Counter(s) for ch in s: remain_counter[ch] -= 1 if ch in stack: # 已经加过的不能再加(贪心:用前面的肯定更小) continue while stack and ch < stack[-1] and remain_counter[stack[-1]] > 0: stack.pop() stack.append(ch) return ''.join(stack)
注意:执行过程中栈不一定是单调,例如1 3 2 4
,栈就是1 3 2 4
,2虽然比3小,但不能把3抛弃,因为后面没有3了。
思路:从前往后,如果新加入的元素比栈顶小,且K还有剩余,就加栈顶元素抛弃。
这也有贪心的思想,如果前i个把K次机会用了,总比留着后面用好,所以不会有影响。
代码实现参考https://leetcode-cn.com/problems/remove-k-digits/comments/667737
class Solution: def removeKdigits(self, num: str, k: int) -> str: stack = [] for d in num: while stack and k and stack[-1] > d: stack.pop() k -= 1 stack.append(d) if k > 0: stack = stack[:-k] return ''.join(stack).lstrip('0') or "0"
注意:执行过程也不一定是单调栈,例如4 3 4 2 5,K=2
,栈会出现3 2 5
,可见不是单调的。后面的两题也是这样。
思路:和上题一样,还不用去前导0。甚至可以直接返回 return self.removeKnum(nums, len(nums)-k);
保留K个相当于删除len(nums)-k
class Solution: def mostCompetitive(self, nums: List[int], k: int) -> List[int]: stack = [] remain = len(nums)-k for num in nums: while stack and num < stack[-1] and remain: stack.pop() remain -= 1 stack.append(num) return stack[:k]
题意:选取\(K\)个最大的,改成大于栈顶时出栈;其次是两次数组,枚举每种情况,数组A取\(i\)个,那么数组B就去\(K-i\)个;后面将两个数组(不一定有序)按队头较大者逐个合并。
代码实现参考https://leetcode-cn.com/problems/create-maximum-number/comments/689718
class Solution: def pickKnum(self, nums, k): stack = [] remove = len(nums)-k; for num in nums: while stack and num > stack[-1] and remove: stack.pop() remove -= 1 stack.append(num) return stack[:k] def merge(self, A, B): nums = [] while A or B: big = A if A>B else B nums.append(big[0]); big.pop(0) # A,B会随着bigger的改变而改变 return nums def maxNumber(self, nums1: List[int], nums2: List[int], k: int) -> List[int]: ans = [0]*k; for i in range(k+1): if i <= len(nums1) and k-i <= len(nums2): # print(self.pickKnum(nums1, i)) # print(self.pickKnum(nums2, k-i)) ans = max(ans, self.merge(self.pickKnum(nums1, i), self.pickKnum(nums2, k-i))) return ans;