Участник:F.Nikitin/AdvantageCount

Материал из DISCOPAL
Перейти к: навигация, поиск
class Solution:
    def advantageCount(self, A, B):
        res = [-1] * len(A)
        A = collections.deque(sorted(A))
        B = collections.deque(sorted((b, i) for i, b in enumerate(B)))
        for _ in range(len(A)):
            a = A.popleft()
            b = B[0]
            if a > B[0][0]:
                B.popleft()
                res[b[1]] = a
            else:
                b = B.pop()
                res[b[1]] = a
        return res