[Python | 프로그래머스] 연속된 수의 합

https://school.programmers.co.kr/learn/courses/30/lessons/120923

프로그래머스

 

문제

연속된 세 개의 정수를 더해 12가 되는 경우는 3, 4, 5입니다. 두 정수 num과 total이 주어집니다. 연속된 수 num개를 더한 값이 total이 될 때, 정수 배열을 오름차순으로 담아 return하도록 solution함수를 완성해보세요.

 

제출 답안

def solution(num, total):
    answer = [9999]
    last = total
    if last == 0: last = num
    while(sum(answer)!=total):
        answer = []
        for i in range(num):
            answer.append(last-i)
        last -= 1
    return sorted(answer)

 

개선 답안

중앙에 위치한 값이 total // num

def solution(num, total):
    avg = total // num
    return [i for i in range(avg - (num-1)//2, avg + (num+2)//2)]

 

참고 자료

https://dduniverse.tistory.com/entry/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EC%97%B0%EC%86%8D%EB%90%9C-%EC%88%98%EC%9D%98-%ED%95%A9-%ED%8C%8C%EC%9D%B4%EC%8D%AC-python

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