본문 바로가기
Algorithm/BFS, DFS

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

by 등촌동 꼬북이 2020. 8. 28.

a와 b가 주어지면 그 사이의 값들을 전부 합해서 return하면 되는 문제

 

b가 a보다 클 경우가 있으므로 if else문을 활용하였고 내장함수인 sum과 range로 해결

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

댓글