[Programmers] 두 정수 사이의 합

두 정수 사이의 합

level 1

연습문제

 

python3

def solution(a, b):
    if a == b:
        return a;

    summery = 0

    if (a > b):
        tmp = b
        b = a
        a = tmp

    for num in range(a, b + 1):
        summery += num

    return summery

range와 for만 잘 쓰면 되는 문제.

def solution(a, b):
    if a > b: a, b = b, a

    return sum(range(a,b+1))

파이썬에서는 tuple을 이용해 스왑을 편하게 할 수 있다. 

sum도 함수로 제공하고 있었다.. 😅