본문 바로가기

Python_Matter/COS Pro

YBM Python Test Level 2 - n부터 m까지 자연수의 합

Q>

문제 설명
두 자연수 n부터 m까지의 합을 구하려고 합니다.
이를 위해 다음과 같이 3단계로 간단히 프로그램 구조를 작성했습니다.

1. 1부터 m까지의 합을 구합니다.
2. 1부터 n-1까지의 합을 구합니다.
3. 1번 단계에서 구한 값에서 2번 단계에서 구한 값을 뺍니다.

두 자연수 n과 m이 매개변수로 주어질 때, n 부터 m 까지의 합을 return 하도록 solution 함수를 작성했습니다. 
이때, 위 구조를 참고하여 중복되는 부분은 func_a라는 함수로 작성했습니다. 
코드가 올바르게 동작할 수 있도록 빈칸을 알맞게 채워주세요.

- 매개변수 설명
두 자연수 n과 m이 solution 함수의 매개변수로 주어집니다.
* n, m은 1 이상 10,000 이하의 자연수이며, 항상 n ≤ m 을 만족합니다.

- return 값 설명
solution 함수는 n부터 m까지의 합을 return 합니다.

- 예시

n m result
5 10 45
6 6 6


- 예시 설명
예시 #1
5부터 10까지 자연수의 합은 45입니다.

- 예시 #2
6부터 6까지 자연수의 합은 6입니다.

 

A>

def func_a(k):
    sum = 0
    for i in range(k + 1):          # 1 부터 k까지 구하므로 k + 1 또는 k, 0, -1 또는 1, k + 1
        sum += i
    return sum

def solution(n, m):
    sum_to_m = func_a(m)
    sum_to_n = func_a(n-1)
    answer = sum_to_m - sum_to_n
    return answer

print(solution(5, 10))
print(solution(6, 6))

 

F>

Quiz1.py
0.00MB

 

#>

https://programmers.co.kr/learn/courses/33