LeetCode 870 Advantage Shuffle

最新推荐文章于 2024-10-09 09:53:23 发布

frankguodongchen

最新推荐文章于 2024-10-09 09:53:23 发布

阅读量91

点赞数

分类专栏： Python学习文章标签：算法 leetcode python 数据结构

本文链接：https://blog.csdn.net/frankguodongchen/article/details/105549708

版权

Python学习专栏收录该内容

137 篇文章 0 订阅

订阅专栏

LeetCode 870

这个问题要返回的是任意一个组合，那么我们考虑，如果我们每次都用最小的大于第二个数组的位置的数就可以，那么可以用二分查找来做。

    def advantageCount(self, A: List[int], B: List[int]) -> List[int]:
        if A == None or B == None: return None
        A = sorted(A)
        result = list()
        for b in B:
            pos = bisect.bisect_right(A, b)
            if pos >= len(A): pos = 0
            if A[pos] == B: pos +=1
            result.append(A[pos])
            A.pop(pos)

        return result