[swea] 5204. 베이비진 게임 / python 파이썬

🚩 그리디

 

thinking

player1과 player2의 카드 결과를 카운팅할 cnt_1, cnt_2 이라는 리스트를 만들고

input 데이터를 하나씩 할당해주면서 babygin 함수로 run과 triplet을 체크해주었다.

 

코드

T = int(input())

def babygin(num, cnt_lst):  # 카드숫자, 카운팅리스트
    cnt_lst[num] += 1
    flag = 0
    i = 0
    while i < 8:
        if cnt_lst[i] >= 3:  # triplet check
            flag = 1
            break
        if cnt_lst[i] and cnt_lst[i + 1] and cnt_lst[i + 2]:  # run check
            flag = 1
            break
        i += 1
    if flag == 1:
        return True


for tc in range(1, T+1):
    card_lst = list(map(int, input().split()))
    cnt_1, cnt_2 = [0]*10, [0]*10

    winner = 0
    for i in range(0, len(card_lst), 2):
        if babygin(card_lst[i], cnt_1):
            winner = 1
            break
        if babygin(card_lst[i+1], cnt_2):
            winner = 2
            break

    print("#{} {}".format(tc, winner))