[Programmers] 최고의 집합

태환·2024년 3월 26일
0

Coding Test

목록 보기
151/151
post-custom-banner

📌 [Programmers] 최고의 집합

📖 문제

📖 예제

📖 풀이

def solution(n, s):
    if n > s:
        return [-1]
    
    initial = s // n
    answer = [initial for _ in range(n)]
    
    idx = len(answer) - 1
    for _ in range(s % n):
        answer[idx] += 1
        idx -= 1
    return answer

n이 s보다 클 경우 각 요소의 값은 0을 가질 수 밖에 업게 되므로 [-1]을 출력한다.
이외의 경우 s // n 값을 각 요소에 초기화해주고 나머지 값은 요소에 1씩 증가시켜주는 방식으로 할당한다.

profile
연세대학교 컴퓨터과학과 석사 과정
post-custom-banner

0개의 댓글