Lv0. 연속된 수의 합

연속된 수의 합

문제 설명

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

제한사항

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

입출력 예

num	total	result
3	12	[3, 4, 5]
5	15	[1, 2, 3, 4, 5]
4	14	[2, 3, 4, 5]
5	5	[-1, 0, 1, 2, 3]

---

내 풀이

import math

def solution(num, total):
    answer = []
    if num%2 == 1:
        mid = total//num
        diff = (num-1)//2
        answer = list(range(mid-diff, mid+diff+1))
    else:
        mid = total/num
        diff = num//2
        answer = list(range(math.ceil(mid-diff), math.ceil(mid+diff)))
        
    return answer

단점이 몇개 있는데 일단은 np나 pd 같은 모듈을 너무 많이 쓰다보니까 일반 파이썬에서 모듈없이 해당 내용을 어떻게 코드화 하는지를 잊어버려서 ^^.. 틀린건 아니지만 모듈을 사용해서 결과값을 도출하려고 하면 오브젝트 형태가 달라지니까 코드가 인식을 못한다 ㅎㅎ.. 기본에 충실해야되는데 기본이 없었음

다른 사람 풀이

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

보고 어라라 싶었던 풀이ㅋㅋㅋㅋ