🥉 14696번: 딱지놀이

👉🏻 14696번: 딱지놀이

import sys; input=sys.stdin.readline

def compare(A, B, value):
    if A.count(value) != B.count(value):
        return 'A' if A.count(value) > B.count(value) else 'B'
    return None

for _ in range(int(input())):
    A = list(map(int, input().split()))[1:]
    B = list(map(int, input().split()))[1:]
    
    result = None
    for value in [4, 3, 2, 1]:
        result = compare(A, B, value)
        if result:
            break
    
    print(result or 'D')

• 단순 구현 문제입니다.
  
  • 중복되는 if-elif 구문을 함수로 정리하여 리팩토링하였습니다.
  • 그러나, 이중 반복문이 생기는 덕에 시간 복잡도는 조금 증가한 듯 합니다.
  • Counter를 사용해서 풀면 count() 함수 호출 횟수를 줄일 수 있어 개선이 가능합니다!