力扣—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]

解题思路

这题和402的解题思路很像，402题是删除k个数使列表最小，而这里是删除m个数使nums1最大，删除n个数使nums2最大，并且m+n = k，可以使用一个循环，从nums1取0个，nums2取k个；从nums1取1个，nums2取k-1个 . . . . . . ，以示例1为例，假设从nums1取出[6, 5]，从nums2取出[9, 8, 3], 合并的时候可以遍历nums1每个元素，对于6，找出nums2第一个比6小的数，把6插入到该数之前，所以把6插入后为[9, 8, 6, 3]，然后再插入5，同理得到[9, 8, 6, 5, 3]，推荐别人写的博客：https://leetcode-cn.com/problems/create-maximum-number/solution/yi-zhao-chi-bian-li-kou-si-dao-ti-ma-ma-zai-ye-b-7/

class Solution(object):
    def pick_max(self, data, i):
        stack = []
        del_num = len(data) - i
        for digit in data:
            while del_num and stack and stack[-1] < digit:
                stack.pop()
                del_num -= 1
            stack.append(digit)
        return stack[0: i]

    def merge(self, data1, data2):
        ans = []
        while data1 or data2:
            bigger = data1 if data1 > data2 else data2
            ans.append(bigger[0])
            bigger.pop(0)
        return ans

    def maxNumber(self, nums1, nums2, k):
        max_res = [-float('inf')]
        for i in range(k + 1):
            if i <= len(nums1) and k - i <= len(nums2):
                temp1 = self.pick_max(nums1, i)
                temp2 = self.pick_max(nums2, k - i)
                max_res = max(max_res, self.merge(temp1, temp2))
        return max_res

S = Solution()
print(S.maxNumber([3, 9], [8, 9], 3))