[15235] Olympiad Pizza

문제보기

해결코드

import sys

n = int(sys.stdin.readline().rstrip())
peice_nums = []
count_nums = [0] * n  # count_nums 리스트를 0으로 초기화
total = 0

x_list = sys.stdin.readline().rstrip().split()
for x in x_list:
    x = int(x)
    peice_nums.append(x)
    total = total + x

cnt=0

while(1):
    for i in range(n):
        if(peice_nums[i]>0):
            peice_nums[i] = peice_nums[i]-1
            cnt+=1
            if(peice_nums[i]==0):
                count_nums[i] = cnt

    if(cnt==total):
        for x in count_nums:
            print(x, end=" ")
        break