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

 

문제 설명

두 정수 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

접근법

중심을 기준으로 양쪽의 끝값이 같은 것을 이용해 보겠습니다. 즉 1, 2, 3, 4, 5는 3을 기준으로 1+5 = 2+4 임을 이용할것입니다!

 

나의 코드

def solution(a, b):
    if a == b : return a
    elif a > b : 
        return ((a+b)*(abs(a-b)//2))+((a+b)//2) if abs(a-b)%2 == 0 else (a+b)*(abs(a-b)//2 + 1)
    else : 
        return ((a+b)*(abs(b-a)//2))+((a+b)//2) if abs(b-a)%2 == 0 else (a+b)*(abs(b-a)//2 + 1)

더해야할 숫자의 개수가 홀 수 일때는 중간값을 더해줘야하며, 짝수일 경우에는 중간값을 더해주지 않아도 됩니다!

 

다른 코드

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

복잡하게만 생각했는데 위와 같이 단순 반복문을 사용하여 푸는 방법도 있습니다!