[알고리즘] 백준 9037 The candy war / python

https://www.acmicpc.net/problem/9037

9037번: The candy war

 

def check(N, C):
    for i in range(N):
        if C[i] %2 == 1:
            C[i] += 1
    return len(set(C)) == 1    # True 또는 False

def share(N, C):
    temp = [0] * N
    for i in range(N):
        if C[i] % 2 == 1:
            C[i] += 1
        C[i] //= 2
        temp[(i+1) % N] += C[i]

    for j in range(N):
        C[j] += temp[j]

def func():
    N, C = int(input()), list(map(int, input().split()))
    count = 0
    while not check(N, C):
        share(N, C)
        count += 1
    print(count)

T = int(input())
for _ in range(T):
    func()

func함수 안에서
처음 순환을 시작할때 홀수인지를 체크하는 check함수와
사탕의 반을 오른쪽에 앉은 아이에게 주는 share함수를 부른다