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

문제 설명

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

 

제한 조건

• a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
• a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
• a와 b의 대소관계는 정해져있지 않습니다.

 

입출력 예

a	b	return
3	5	12
3	3	3
5	3	12

 

Solution

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

 

코드해석

입력받은 a, b중 최솟값과 최댓값을 min, max 함수를 이용하여 찾습니다. 최솟값부터 최댓값까지의 정수를 for 반복문을 사용하여 더하고 결과를 반환합니다.

 

다른 사람의 풀이

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

 

a, b의 대소관계가 정해져있기 때문에 어떤 수가 더 큰지 알 수 없죠. 그래서 먼저 입력받은 a가 b보다 크다면 두 숫자를 교환하고, sum 함수를 사용하여 range(a, b+1)에 해당하는 값들을 한번에 더합니다.