Python 두 정수 사이의 합 - Programmers

코딩테스트 연습 두 정수 사이의 합

---

두 정수 사이의 합

문제 설명

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요. 예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

제한 사항

• a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
• a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
• a와 b의 대소관계는 정해져있지 않습니다.

입출력 예

a = 3
b = 5
return = 12

a = 5
b = 3
return = 12

코드

def solution(a, b):
    answer = 0
    if (a > b) : a, b = b, a
    for i in range(a, b+1): answer += i
    return answer

고찰

python에서는 대수 비교하여 값을 변경하는 경우 간단히 한 줄로 표현할 수 있다. 하지만 c++과 마찬가지로 수학적으로 접근하는 방법이 중요한 것 같다. 수열을 사용해서 풀면 훨씬 빠른 속도로 프로그램을 돌릴 수 있다.