[프로그래머스] 두 정수 사이의 합 (Python)

문제 링크 (Level 1)

https://programmers.co.kr/learn/courses/30/lessons/12912?language=python3 

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

 

나의 코드 (Python)

def solution(a, b):
    answer = 0

    if a > b:   # 대소비교
        a, b = b, a

    for i in range(a, b+1): # 두 수 사이 합 구하기
        answer += i
        
    return answer

 

먼저 a, b는 대소관계가 없으므로 대소비교를 먼저 합니다.

이후 두 수 사이 합을 구합니다.