본문 바로가기

알고리즘 문제

Programmers Level 4 최고의 집합

#s - s//n 를 원소에 배분한다.

def bestSet(n, s):

    if n>s:

        return [-1]

    answer = []

    avg = s//n

    for i in range(n):

        answer.append(avg)

    addingNumber = s-avg*n

    while addingNumber>0:

        for i in range(n):

            if addingNumber>0:

                answer[i]+=1

                addingNumber -= 1

            else: break

    return sorted(answer)


# 아래는 테스트로 출력해 보기 위한 코드입니다.

print(bestSet(3,13))