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

프로그래머스

Lv.1 두 정수 사이의 합

1.문제

input 두 정수
output 사이의 합 return, 단 a, b 사이 대소관계 X

2. 풀이

range로 for문 돌림

3. 코드

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

모범코드

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

반성)

• range와 sum 이용하기
• if else 한줄
• 만약 값 변경한다면 a,b = b,a 가능