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

---

문제설명

https://programmers.co.kr/learn/courses/30/lessons/12912

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

사고과정

• 단순이 리스트 컴프리헨션을 사용한 것보다 제네레이터를 활용해서 구현했다.

풀이

def solution(a, b):
    if a == b:
        return a
    elif a < b:
        return sum(x for x in range(a, b+1))
    else:
        return sum(x for x in range(b, a+1))