LeetCode321. 拼接最大数（单调栈、分治）

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、代码详解

https://leetcode-cn.com/problems/create-maximum-number/solution/yi-zhao-chi-bian-li-kou-si-dao-ti-ma-ma-zai-ye-b-7/

单调栈+分治

输入:
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))