공부, 기록

프로그래머스 최고의 집합 파이썬(PYTHON) 본문

코딩

프로그래머스 최고의 집합 파이썬(PYTHON)

무는빼주세요 2020. 11. 14. 13:50

문제링크 : programmers.co.kr/learn/courses/30/lessons/12938

 

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

자연수 n 개로 이루어진 중복 집합(multi set, 편의상 이후에는 집합으로 통칭) 중에 다음 두 조건을 만족하는 집합을 최고의 집합이라고 합니다. 각 원소의 합이 S가 되는 수의 집합 위 조건을 만족

programmers.co.kr

 

 

import heapq
def solution(n, s):
    answer = []
    if n>s:
        return [-1]
    a,b = divmod(s,n)
    answer=[a for i in range(n)]
    idx=len(answer)
    while b>0:
        answer[idx-1]+=1
        idx-=1
        b-=1
    return answer