LeetCode题解(0870)：优势洗牌(Python)

最新推荐文章于 2022-10-08 16:48:23 发布

长行

最新推荐文章于 2022-10-08 16:48:23 发布

阅读量316

点赞数

分类专栏：刷题记录文章标签： leetcode python 数组贪心算法排序

本文链接：https://blog.csdn.net/Changxing_J/article/details/112782981

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

这篇博客探讨了一种利用贪心算法和排序策略解决数组中找优势数的问题。代码实现包括Python解法，主要涉及时间复杂度为O(NlogN)，空间复杂度为O(N)的算法。通过建立映射和移除无效元素，实现了找到数组A中比数组B中对应元素大的数并将它们重新排列的目标。

摘要由CSDN通过智能技术生成

题目：原题链接（中等）

标签：数组、贪心算法、排序

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O (N l o g N)$	$O (N)$	384ms (84.62%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

class Solution:
    def advantageCount(self, A: List[int], B: List[int]) -> List[int]:
        size = len(A)
        sorted_A, sorted_B = sorted(A), sorted(B)

        mapping = collections.defaultdict(list)
        useless = set()
        i1, i2 = 0, 0
        while i1 < size:
            while i1 < size and sorted_A[i1] <= sorted_B[i2]:
                useless.add(sorted_A[i1])
                i1 += 1
            if i1 < size:
                mapping[sorted_B[i2]].append(sorted_A[i1])
            i2 += 1
            i1 += 1

        ans = []
        for b in B:
            if mapping[b]:
                ans.append(mapping[b].pop())
            else:
                ans.append(useless.pop())
        return ans