[프로그래머스] Lv.1 두 정수 사이의 합 (파이썬)

a부터 b까지 합을 구하는 단순한 문제

def solution(a, b):
    answer = 0
    if a<b:     #b가 a보다 크면 a부터
        for i in range(a,b+1):
            answer+=i
    else:       #a가 b보다 크면 b부터
        for i in range(b,a+1):
            answer+=i
    return answer