[Python] 최고의 집합

코딩테스트 연습 - 최고의 집합

풀이

1. n이 s보다 크면 집합을 만들 수 없으므로 [-1] 리턴

2. s를 n으로 나눈 몫을 n개 만큼 answer에 append

3. s를 n으로 나눈 나머지를 1씩 뒷쪽부터 더해준다.

 

def sol(n, s):
	answer = []
    if n > s: return[-1]
    
    v = s // n
    for i in range(n):
    	answer.append(v)
    
    idx = len(answer) - 1
    for i in range(s % n):
    	answer[idx] += 1
        idx -= 1
    
    return answer