给定两个大小相等的数组 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]