[ 파이썬(python) ] 프로그래머스 level1 - 두 정수 사이의 합

📌 두 정수 사이의 합

---

💡 나의 풀이

문제를 보고 3가지 경우의 수를 판단했다.

1. a == b
2. a < b
3. a > b

sum(range())를 사용하면 간편하게 줄일 수 있었는데, 무심코 반복문을 사용해서 효율성을 극대화시키지 못할 뻔했다.

a가 b보다 크면, swap처리를 해준다. 이후, sum(range())를 사용하여 값을 계산해준다.

a와 b가 같으면, 0아니야? 라고 할 수도 있지만 마지막 값은 +1을 해줬기 때문에 입력값 그대로 리턴되는것을 알 수 있다.

def solution(a, b):
    if a > b:
        a, b = b, a

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