给定两个大小相等的数组 A 和 B,A 相对于 B 的优势可以用满足 A[i] > B[i] 的索引 i 的数目来描述。
返回 A 的任意排列,使其相对于 B 的优势最大化。
示例 1:
输入:A = [2,7,11,15], B = [1,10,4,11]
输出:[2,11,7,15]
示例 2:
输入:A = [12,24,8,32], B = [13,25,32,11]
输出:[24,32,8,12]
贪心+穷举
链接:https://leetcode.com/problems/advantage-shuffle/
class Solution(object):
def advantageCount(self, A, B):
"""
:type A: List[int]
:type B: List[int]
:rtype: List[int]
"""
sortedA = sorted(A)
sortedB = sorted(B)
h = {b: [] for b in B}
remain = []
j = 0
for a in sortedA:
if a > sortedB[j]:
h[sortedB[j]].append(a)
j += 1
else:
remain.append(a)
res = []
for b in B:
if h[b]:
res.append(h[b].pop())
else:
res.append(remain.pop())
return res