프로그래머스 LV.1 - 두 정수 사이의 합[Python]

문제 설명

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

def solution(a, b):
    if a < b :
        return sum(list(range(a, b+1)))
    else:
        return sum(list(range(b, a+1)))

풀이

• 두 정수 a,b가 주어졌을 때, a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
• a와 b가 같은 경우는 둘 중 아무 수나 리턴하라.
  -> if 조건문 + sum() + list + range로 구현하였다.

# 짧은 코드
def adder(a, b):
    # 함수를 완성하세요
    if a > b: a, b = b, a

    return sum(range(a,b+1))