num
과 total
을 입력으로 받고, num
개 숫자의 합이 total
을 만족하는 result
배열을 리턴하는 문제다.
다행히도 문제를 보자마자 total/num
한 값이 result
배열의 중간값에 해당할 것을 예상했고, num
이 양수일 때와 음수일 때를 구분하여 for
문을 작성했다.
def solution(num, total):
answer = []
mid = int(total / num)
if num % 2 == 1:
for i in range(mid-(int(num/2)), mid+(int(num/2))+1):
answer.append(i)
else:
for j in range(mid-(int(num/2)-1), mid+(int(num/2))+1):
answer.append(j)
return answer
p.s 프로그래머스 ui
가 익숙하지 않아서인지, level 0
인데도 백준 실버~골드 급으로 느껴지는건 무엇일까..