[HackerRank] Cut the Sticks

[문제 링크]

[입력]

int arr[n]: 각 막대기의 길이를 담은 배열 , 크기는 n 이다.

[출력]

int[]: 각 반복시 남은 막대기의 개수를 담은 배열

[코드]

def cutTheSticks(arr):
    # Write your code here
    result = list()

    while 1:
        if(len(arr)==0):
            return result
        else:
            temp = list()
            result.append(len(arr))
            length_of_cut = min(arr)
            for idx in range(len(arr)):
                if arr[idx] == length_of_cut:
                    arr[idx] = 0
                else:
                    arr[idx] -= length_of_cut
                
                if(arr[idx] !=0):
                    temp.append(arr[idx])
            arr = temp