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

 

def solution(a, b):
    if a == b:
        return a
    else:
        answer = 0

        for i in range(min(a,b), max(a,b)+1):
            answer += i

    return answer

 

def solution(a, b):
    if a == b:
        return a
    else:
        return sum(range(min(a,b), max(a,b)+1))

 

range()를 통해 a와 b 사이에 속한 모든 정수를 구한다

range(a, b)면 a부터 b-1까지 구하므로 b+1을 해야한다

이 때 b가 더 커야하므로 min(), max()를 통해 더 작고 큰 것을 구한다

 

sum()

- iterable의 합을 구하는 함수