[Python] 숫자 게임

코딩테스트 연습 - 숫자 게임

풀이

1. A의 최솟값이 B의 최댓값 보다 크면 B팀은 절대 못 이김

2. A를 역순으로 정렬.

3. B의 큰 값부터 비교하기 위해 heapq 사용

 

import heapq

def solution(A, B):
    if min(A) > max(B):
        return 0
    
    A.sort(reverse=True)
    B = [-i for i in B]
    heapq.heapify(B)
    
    cnt = 0
    for a in A:
        if a >= abs(B[0]):
            continue
        else:
            heapq.heappop(B)
            cnt += 1
    
    return cnt