拼接最大数

最新推荐文章于 2024-12-07 17:24:17 发布

转载最新推荐文章于 2024-12-07 17:24:17 发布 · 333 阅读

CC 4.0 BY-SA版权

原文链接：https://leetcode-cn.com/problems/remove-duplicate-letters/solution/yi-zhao-chi-bian-li-kou-si-dao-ti-ma-ma-zai-ye-b-4/

文章标签：

#leetcode #算法 #python

每天一个Python小技巧专栏收录该内容

33 篇文章

订阅专栏

博客围绕 LeetCode 12.2 每日一题展开，题目是从两个数组中选 k 个数字拼接成最大数，且同一数组数字保持相对顺序。介绍基础思想是从两数组分别选数再拼接，还通过具体示例展示如何在数组中找最大组合，要综合考虑数组大小和选取数量。

LeetCode每日一题（12.2）

#困难#
给定长度分别为 m 和 n 的两个数组，其元素由 0-9 构成，表示两个自然数各位上的数字。现在从这两个数组中选出 k (k <= m + n) 个数字拼接成一个新的数，要求从同一个数组中取出的数字保持其在原数组中的相对顺序。
求满足该条件的最大数。结果返回一个表示该最大数的长度为 k 的数组。
链接：https://leetcode-cn.com/problems/create-maximum-number

关于查找最大数和最小数.
https://leetcode-cn.com/problems/remove-duplicate-letters/solution/yi-zhao-chi-bian-li-kou-si-dao-ti-ma-ma-zai-ye-b-4/

基础思想是，从nums1找i个，nums2找k-i个. 然后在把两组数拼接. 注意保持相对位置不变，并不是只能两个数据集前后拼接，可以穿插.（这就不好做了）

nums1=[3,4,6,5]
nums2=[9,1,2,5,8,3]
先分别找最大组合. 遍历数组，判断是保留还是去除.
构建空box来存储
以i=2为例（即在nums1中选取2个数）
a. 3 第一次为空数组，所以直接存入.
b. 4>3，去掉3，存入4. 用pop实现
c. 6>4，去掉4，存入6
d. 5<6，由于i=2，所以只能存入
最后box=[6,5]
k=5，那么在nums2中就该选择k-i=3个数.
a. 9
b. 1<9，丢弃1，
c. 2<9，丢弃2
d. 5<9，丢弃5
e. 8<9，由于必须选择3个数，因此存入8
f. 3<9，存入3
最后box=[9,8,3]
很明显，既要判断原数组和新数组的大小，是否填入，还要考虑选取的总数量，确保一定要选出这么多数. 可以考虑将所有的数全部选出来，然后截取要的 i or k-i 个数

class Solution:
    def maxNumber(self, nums1: List[int], nums2: List[int], k: int) -> List[int]:
        def pick_max(nums, k):
            box = []
            stop = len(nums) - k
            for i in nums:
                while stop and box and box[-1] < i:
                    box.pop()
                    stop -= 1
                box.append(i)
            return box[:k]
        def merge(A, B):
            D = []
            while A or B:
                bigger = A if A > B else B
                ans.append(bigger[0])
                bigger.pop(0)
            return D
        return max(merge(pick_max(nums1, i), pick_max(nums2, k-i)) for i in range(k+1) if i <= len(nums1) and k-i <= len(nums2))