leetcode 870 优势洗牌

leetcode 870

题目描述：给定两个大小相等的数组 A 和 B，A 相对于 B 的优势可以用满足 A[i] > B[i] 的索引 i 的数目来描述。
返回 A 的任意排列，使其相对于 B 的优势最大化。

我的思路：这道题思路蛮好想的，就是每次贪心的找A中最小的大于当前B中元素的元素，听起来有点绕？那给你四个字：田忌赛马……是的，这是同一个问题。但这下就卡在了python的实现上，在C++里我们有lower-bound和upper-bound，但在python里我们没有，所以我们就想到了类似于双指针的解法。写了半天发现自己还是太菜了，于是看了官方题解，结果发现题解的思路是不同的！

题解思路：在我的思路中，我们会自然地把A排序，然后遍历B，对A开一个指针。但是题解告诉我们，我们可以对A、B同时排序。然后怎么处理呢？首先，我们开一个字典，键值为B中元素，值为一个列表。然后我们按位置同时遍历A 和B，也就是使A、B索引值相同，如果A[i]>B[i]，当前A[i]就加入B[i]的列表中，否则，就加入一个新的待定列表。最后构造答案时，对于B中的每个元素，列表有元素就输出，没有就从待定列表中拿。

class Solution:
    def advantageCount(self, A: List[int], B: List[int]) -> List[int]:
        a = sorted(A)
        b = sorted(B)
        ans = {item: [] for item in B}
        rest = []
        i = 0
        for item in a:
            if item > b[i]:
                ans[b[i]].append(item)
                i += 1
            else:
                rest.append(item)
        res = []
        for item in B:
            if ans[item]:
                res.append(ans[item].pop())
            else:
                res.append(rest.pop())
        return res

考虑这样贪心为什么是对的，因为两个列表都经过排序后，如果A[i]>B[i]，则A[i]一定是A中大于B[i]的最小值，这个A[i]也就是我们想要的答案。而如果A[i]<=B[i]，那A[i]肯定更小于等于B[i]之后的元素，也就是这个A[i]没用了。

最后还有一点，可能有人会好奇为啥字典里的值要是一个列表，直接是一个数不是更简单么？当然不是，因为B中数据是会有重复的！会重复！