870. 优势洗牌

leetcode 870. 优势洗牌

题目描述

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

解题思路

类似于田忌赛马。使用贪心算法，每次找出A中最小牌来配对B中比他小的，这时候B中的不一定要是最小的，假如此时A中最小为a，那么A剩下的牌对可以与a配对的牌都是具有优势的，若a没有配对的，即B中没有比他小的，那么将该值保存在临时数组temp_num中，等所有的数配对完再将其插入到还空了的位置。
所以初步想法是将A进行排序，从小到大依次取出，假如取出a，再将a与B中的每个数进行比较，如果碰到比a小的，则将a填入该位置。接着配对下一个数。
初步代码如下：

class Solution:
    def advantageCount(self, A: List[int], B: List[int]) -> List[int]:
        A.sort()	# 对A排序
        temp_num = []      # 保存A中配对不成功的数 
        adv_A = [None] * len(A)
        B = list(enumerate(B))
        for i in range(len(A)):
            flag = True
            for b in B:
                if A[i]>b[1] and adv_A[b[0]]==None:
                    adv_A[b[0]] = A[i]   # 将匹配到具有优势的数填入相应位置
                    flag = False
                    B.remove(b) # 将配对过的数删除，避免重复比较
                    break
            if flag:
                temp_num.append(A[i])  # 如果a在B中不存在优势配对的数，则保存到temp_num中
        # print(temp_num)
        m = 0
        for t in range(len(A)):
            if adv_A[t]==None:
                adv_A[t] = temp_num[m]  # 将配对不成功的数填入到空缺的地方
                m += 1

        return adv_A

思路是对的，不过代码时间复杂度感人，应该还可以救一救。
1、 将B也排序，不过排序前也要保留B中元素原始的index。那么此时a配对的就是B中最小牌b（除去B中之前已经配对过的）

B = list(enumerate(B))
B.sort(key=lambda x:x[1])

2、 两层循环会多次重复比较之前比较过的数。
3、 可以只循环n次。设置一个指向B中当前配对元素b的指针，每次找A中比他大的最小的a

最终代码

class Solution:
    def advantageCount(self, A: List[int], B: List[int]) -> List[int]:
        A.sort()
        temp_num = []      
        adv_A = [None] * len(A) # 排序后的优势数组
        # 对B进行排序，并且保留B中元素原始index
        B = list(enumerate(B))
        B.sort(key=lambda x:x[1])
        t = 0
        for i in range(len(A)): # 循环n次
            if A[i]>B[t][1]: # 如果a比当前指向的b大则保留，
                adv_A[B[t][0]] = A[i]
                t += 1 

            else:
                temp_num.append(A[i])# 否则保存到temp_num，等全部匹配完成再将它填入空缺的地方

        m = 0
        for j in range(t, len(A)):
            adv_A[B[j][0]] = temp_num[m]
            m += 1

        return adv_A