[파이썬] 프로그래머스 숫자 게임

오랜만에 3단계를 맞췄다. 두 리스트 정렬 후 풀었고, 시간 초과가 계속 났으나 count 변수를 생성해줘서 이를 해결했다.

 

import heapq
 
 
def solution(A, B):
    answer = 0
 
    # 숫자가 큰 쪽이 승리
 
    # A[i] 보다 최소로 큰 숫자 return, 없으면 최솟값 return
 
    A = sorted(A)
    B = sorted(B)
 
    count = 0
 
    for i in range(0,len(A)):
        for j in range(count,len(B)):
            if B[j]>A[i]:
                answer+=1
                count = j+1
 
                break
 
 
 
 
 
 
    return answer