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

첫 코드

def solution(a, b):
    x = []
    if (a < b):
        for i in range(a, b+1):
            x.append(i)
        answer = sum(x)
    elif (b < a):
        for i in range(b, a+1):
            x.append(i)
        answer = sum(x)
    else:
        answer = a
    return answer

개선된 코드

def solution(a, b):
    answer = 0
    
    if a > b:
        answer = sum(range(b, a + 1))
    elif b > a:
        answer = sum(range(a, b + 1))
    elif a == b:
        return a
    
    return answer

처음에는 불필요하게 for문을 추가하였다. 복잡도 면에서 간결한 알고리즘으로 개선하였다.