1、题目描述
给定长度分别为 m 和 n 的两个数组,其元素由 0-9 构成,表示两个自然数各位上的数字。
现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数,要求从同一个数组中取出的数字保持其在原数组中的相对顺序。
求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。
输入:
nums1 = [6, 7]
nums2 = [6, 0, 4]
k = 5
输出:
[6, 7, 6, 0, 4]
输入:
nums1 = [3, 9]
nums2 = [8, 9]
k = 3
输出:
[9, 8, 9]
2、代码详解
单调栈+分治
输入:
nums1 = [3, 4, 6, 5]
nums2 = [9, 1, 2, 5, 8, 3]
k = 5
输出:
[9, 8, 6, 5, 3]
假如我们从 num1 中取出 1 个数字为 A数组,那么就要从 nums2 中取出 4 个数字为 B数组。
- 运用402题的方法,我们计算出应该取 nums1 的 [6],并取 nums2 的 [9,5,8,3]。 402. 移掉K位数字(单调栈)https://blog.csdn.net/IOT_victor/article/details/110493960
如何将 [6] 和 [9,5,8,3],使得数字尽可能大,并且保持相对位置不变呢?
- 实际上这个过程有点类似归并排序中的治,而上面我们分别计算 num1 和 num2 的最大数的过程类似归并排序中的分。
将 A数组 和 B数组 按照上面的 merge 方法合并
暴力了 k 种组合情况,只需要将 k 种情况取出最大值即可。
class Solution(object):
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 or B:
bigger = A if A > B else B
ans.append(bigger.pop(0))
return ans
max_num_list = []
for i in range(k+1):
if i <= len(nums1) and k-i <= len(nums2):
max_num_list = max(max_num_list, merge(pick_max(nums1, i), pick_max(nums2, k - i)))
return max_num_list
nums1 = [3, 4, 6, 5]
nums2 = [9, 1, 2, 5, 8, 3]
k = 5
s = Solution()
print(s.maxNumber(nums1, nums2, k))
时间复杂度:
- pick_max 的时间复杂度为 O(M+N) ,其中 M 为 nums1 的长度,N 为 nums2 的长度。
- merge 的时间复杂度为 O(k),再加上外层遍历所有的 k 中可能性。
- 因此总的时间复杂度为 O(k^2 * (M + N))
空间复杂度:使用了额外的 stack 和 ans 数组,因此空间复杂度为 O(max(M,N,k))