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

문제

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

프로그래머스

 

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

 

제한사항

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

 

아이디어

 

total을 num으로 나누면 평균이 되고 평균이 연속된 숫자 리스트의 가운데에 오게 된다. 

홀, 짝에 상관없이 시작과 끝을 구하기 위해 num-1, num+2를 해준다.

 

풀이

 

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