🔗 문제 링크

https://programmers.co.kr/learn/courses/30/lessons/12912

💻 코드

def solution(a, b):
    answer = 0

    if a < b:
        for i in range(a, b + 1):
            answer += i
    else:
        for i in range(b, a + 1):
            answer += i
    return answer

📝 정리

for문 돌려서 합 구하지 말고 sum() 메서드 사용하기

def solution(a, b):
    answer = 0

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