每天坚持刷题!!!
leetcode 870 优势洗牌
题目描述:
给定两个大小相等的数组 A 和 B,A 相对于 B 的优势可以用满足 A[i] > B[i] 的索引 i 的数目来描述。返回 A 的任意排列,使其相对于 B 的优势最大化。
题目分析:
这道题乍看起来似乎可以用动态规划,但是细细一想这道题是无法拆分成子问题的。OK,既然动态规划没法用又是求极值的话,首先尝试用贪心法。
具体分析下,如果给定一个B中的一个数 Bx B x ,贪心的来看我们最好在A中找到所有比这个 Bx B x 大的数中最小的那个,即min([d for d in A if d > Bx B x ]), 这样可以让A中更大的数去匹配B中可能存在的比 Bx B x 大的数
但是对每个B中的数去找这个备选的list实在太浪费时间,所以一个比较好的方法就是先对A和B进行升序排序,两个指针分别指向A和B开头的数,然后A的指针从头开始移动找到第一个比B的指针指向的数大的数,然后记下A的指针指向的数要移动的位置,即要移动到B的指针指向的数的原来的下标,然后B的指针移动到下一个数,直到A或B的指针指向末尾,没有匹配上的数字随机分配。
def advantageCount(self, A, B):
"""
:type A: List[int]
:type B: List[int]
:rtype: List[int]
"""
if not A and not B:
return
if len(A) == len(B) == 1:
return A
bb = sorted([(i, id) for id, i in enumerate(B)], key=lambda x: x[0])
A = sorted(A)
result = [-1 for _ in B]
temp = []
start_a = 0
for data in bb:
while start_a < len(A) and A[start_a] <= data[0]:
temp.append(A[start_a])
start_a += 1
if start_a >= len(A):
break
result[data[1]] = A[start_a]
start_a += 1
for i in xrange(len(result)):
if result[i] == -1:
result[i] = temp.pop()
return result