[Level.0] 연속된 수의 합

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

제한사항

• 1 ≤ num ≤ 100
• 0 ≤ total ≤ 1000
• num개의 연속된 수를 더하여 total이 될 수 없는 테스트 케이스는 없습니다.

문제 해결

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

 

주석: total = (num * (2 * start + num - 1)) / 2 == start = (2*total//num - num+1)//2,  등차수열의 합 공식을 변형하여 첫 번째 항을 구한 뒤 range 함수로 연속된 수열을 생성