字符串:优势洗牌

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

    示例一:输入:A = [2,7,11,15], B = [1,10,4,11],输出:[2,11,7,15]

    示例二:输入:A = [12,24,8,32], B = [13,25,32,11],输出:[24,32,8,12]

    思路:想要使A[i] > B[i]的索引数目最多,应该先考虑大的元素,再考虑小的元素,因为越大的元素越难满足条件。而对于A中的元素,大的元素应该先去和B中的大元素比较,从大到小,找到它“有优势”的元素。因此,肯定是要对两个数组都进行排序的。

    方法一:利用优先队列,把(B[i], i)元组压进队列,自动排序后,根据B的元素大小进行倒序输出,并得到对应的索引值。

    此方法因为用到优先队列,速度并不快

def advantageCount(A, B):
    import Queue as Q

    n = len(A)
    res = [None] * n
    pq = Q.PriorityQueue()

    A.sort()
    for i in range(n):
        pq.put((-B[i], i)) #注意是-B[i],因为需要从大到小输出

    lo = 0
    hi = n - 1
    for i in range(n):
        val, index = pq.get()
        if A[hi] > -val: #把B[i]再变为原值
            res[index] = A[hi] #如果A中的最大值比B[index]大,则该最大值放在index的位置
            hi -= 1
        else:
            res[index] = A[lo] #否则,A中剩下的元素中没有比B[index]大的元素了,把最小值放入该位置
            lo += 1

    return res

    方法二:大神级代码,先把A和B排序,再利用collection,以B中元素b为key,把A中大于B[i]且小于B[i-1]的元素存入key为b对应的list(考虑到B中元素有重复的情况,如果没有重复,那么一个b对应的list也只有一个元素)。如果collection中b对应的list有元素,那么结果中b对应的位置就应该放置该元素,否则,说明该位置没有满足要求的元素,在A剩下的元素中随便取一个即可。

def advantageCount(A, B):
    A = sorted(A)
    take = collections.defaultdict(list)
    for b in sorted(B)[::-1]:
        if b < A[-1]: take[b].append(A.pop())
    return [(take[b] or A).pop() for b in B]

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值