[SWEA 5203번] 베이비진 게임 - 파이썬(python)

SW Expert Academy

문제

0부터 9까지인 숫자 카드 4세트를 섞은 후 6개의 카드를 골랐을 때, 연속인 숫자가 3개 이상이면 run, 같은 숫자가 3개 이상이면 triplet이라고 한다.

게임을 시작하면 플레이어1과 플레이어 2가 교대로 한 장 씩 카드를 가져가며, 6장을 채우기 전이라도 먼저 run이나 triplet이 되는 사람이 승자가 된다.

두 사람이 가져가게 되는 순서대로 12장의 카드에 대한 정보가 주어졌을 때 승자를 알아내는 프로그램을 작성하시오. 만약 무승부인 경우 0을 출력한다.

예를 들어 9 9 5 6 5 6 1 1 4 2 2 1인 경우, 플레이어 1은 9, 5, 5, 1, 4, 2카드를, 플레이어2는 9, 6, 6, 1, 2, 1을 가져가게 된다.

이때는 카드를 모두 가져갈 때 까지 run이나 triplet이 없으므로 무승부가 된다.

입력

첫 줄에 테스트케이스의 수 T가 주어진다. 1<=T<=50

다음 줄부터 테스트 케이스의 별로 각 줄에 0에서 9사이인 12개의 숫자가 주어진다.

출력

각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 답을 출력한다.

풀이

• 플레이어 2명이 12장의 카드를 순서대로 가져가면서 먼저 run이나 triplet이 나온 사람이 이기는 게임
• run이나 triplet이 되려면 플레이어가 최소 3장의 카드를 가져야 하므로 그 때부터 함수를 이용하여 검사한다.
• run_triplet함수에서는 리스트를 정렬하고 run과 triplet을 탐색한다.
• 존재하면 1을 아니면 0을 반환하고 플레이어 중 한명이라도 값이 1이 되면 탐색을 종료한다.
• 탐색이 종료되었을 때 두 플레이어의 값이 같다면 0을 아니라면 값이 1인 플레이어를 출력한다.

코드

def run_triplet(lst):
    lst.sort()
    for i in range(len(lst) - 2):
        if lst[i] == lst[i + 1] == lst[i + 2]:
            return 1

    tri = list(set(lst))
    if len(tri) < 3: return
    for i in range(len(tri) - 2):
        if tri[i] == tri[i + 1] - 1 == tri[i + 2] - 2:
            return 1

    return 0

T = int(input())
for test_case in range(1, T + 1):
    card = list(map(int, input().split()))
    player1 = []
    player2 = []
    first,second=0,0

    for i in range(0,12,2):
        player1.append(card[i])
        player2.append(card[i+1])
        if i>=4:
            first=run_triplet(player1)
            second=run_triplet(player2)
            if first==1 or second==1:
                break

    print(f'#{test_case}',end=' ')
    if first==second:
        print(0)
    elif first==1:
        print(1)
    else:
        print(2)