18. 두 정수 사이의 합

유형 : 구현

풀이 시간 : 5분

 

def solution(a, b):
    answer = 0
    step = 1
    if a > b:
        step = -1
    for i in range(a, b+step, step):
        answer += i
    return answer

 

a, b의 대소관계가 정해져 있지 않다는 점이 포인트인 문제였다. 

전에 <공원 산책> 에서 이것을 고려하지 않아서 한참 헤맸던 기억이 나서, 이번에는 쉽게 해결했다. 

a > b 라면, step을 -1로 설정하여 쉽게 해결하였다.