[프로그래머스 - COS PRO 파이썬 2급] 1. n부터 m까지 자연수의 합

프로그래머스

---

 

문제 설명

두 자연수 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 합니다.

 

예시

예시 설명

예시 #15부터 10까지 자연수의 합은 45입니다.

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

 

---

코드

def func_a(k):
    sum = 0
    for i in range(k):    # 빈칸
        sum += i+1        # 빈칸
		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

 

풀이

• range(k)의 경우, 0부터 k-1까지를 나타냄
• 이 문제에서는 1부터 k까지의 합을 원함
• sum에 값을 하나씩 더할 때, range(k)에서 하나씩 나온 값에서 +1을 한 값을 추가하고자 함