[SWEA] 베이비진 게임

[SWEA] 베이비진 게임

Problem URL : 베이비진 게임

def is_babygin(nums):
    for i in range(8):
        if nums[i] >= 3:
            return True
        if nums[i] and nums[i+1] and nums[i+2]:
            return True
    if nums[8] >= 3 or nums[9] >= 3:
        return True
    return False



for tc in range(1, T + 1):
    N = list(map(int, input().split()))
    ans = 0
    nums_list = [[0] * 10, [0] * 10]
    nums_list[0][N[0]] += 1
    nums_list[0][N[2]] += 1
    nums_list[1][N[1]] += 1
    nums_list[1][N[3]] += 1
    for i in range(4, 12):
        idx = i % 2
        nums_list[idx][N[i]] += 1
        if is_babygin(nums_list[idx]):
            ans = idx + 1
            break

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

Comments

간단한 구현 문제
깔끔하게 코드 짜기가 생각보다 귀찮은 문제인거 같다.