알고리즘

[프로그래머스, 파이썬] 연속된 수의 합

hminor 2024. 2. 23. 09:47
반응형

풀이

  • total%num의 값이 0인지 아닌지 조건분기 후에
  • moc값을 기준으로 값을 하나씩 추가한 뒤 
  • 정렬한 배열의 값을 전달하여 해결

 

def solution(num, total):
    answer = []
    moc = total//num
    if total%num:
        for i in range(num//2):
            answer.append(moc-i)
            answer.append(moc+i+1)
    else:
        for i in range(num//2+1):
            answer.append(moc-i)
            if i != 0: answer.append(moc+i)
           
    return sorted(answer)