321. 拼接最大数(单调栈)
难度困难
给定长度分别为 m
和 n
的两个数组,其元素由 0-9
构成,表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n)
个数字拼接成一个新的数,要求从同一个数组中取出的数字保持其在原数组中的相对顺序。
求满足该条件的最大数。结果返回一个表示该最大数的长度为 k
的数组。
说明: 请尽可能地优化你算法的时间和空间复杂度。
示例 1:
输入: nums1 =[3, 4, 6, 5]
nums2 =[9, 1, 2, 5, 8, 3]
k =5
输出:[9, 8, 6, 5, 3]
示例 2:
输入: nums1 =[6, 7]
nums2 =[6, 0, 4]
k =5
输出:[6, 7, 6, 0, 4]
示例 3:
输入: nums1 =[3, 9]
nums2 =[8, 9]
k =3
输出:[9, 8, 9]
class Solution: def maxNumber(self, nums1, nums2, k): def pick_max(nums, k): stack = [] drop = len(nums) - k for num in nums: while drop and stack and stack[-1] < num: stack.pop() drop -= 1 stack.append(num) return stack[:k] def merge(A, B): ans = [] while A and B: if A > B: ans.append(A.pop(0)) else: ans.append(B.pop(0)) while A: ans.append(A.pop(0)) while B: ans.append(B.pop(0)) return ans max_res = [] for i in range(k+1): if i <= len(nums1) and k-i <= len(nums2): max_res = max(max_res, merge(pick_max(nums1, i), pick_max(nums2, k-i))) return max_res