[Python] 코딩테스트 연습 : 연속된 수의 합

 

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

 

---

 

문제

 

---

 

코드

# 연속된 숫자 num개를 더해서 total을 구하는 숫자 num개를 구하기
def solution(num, total):
    answer = []
    mid_num = total // num
    left_val = num % 2
    # print(mid_num)
    # print("left_val:", left_val)

    answer.append(mid_num)

    left_cnt = num // 2
    # print(left_cnt)

    for i in range(left_cnt):
        left_num = mid_num-(i+1)
        right_num = mid_num+(i+1)
        answer.append(left_num)
        answer.append(right_num)

    answer = sorted(answer)
    if left_val == 0:
        answer.pop(0)


    return answer

 

 

 

---