하루에 하나-9

정인창·2021년 9월 20일
0

코딩테스트 연습

목록 보기
9/19

def solution(a, b):
    if a <= b:
        answer = sum([i for i in range(a, b+1)])
    else:
        answer = sum([i for i in range(b, a+1)])
    return answer
  • a가 b보다 큰 경우가 존재하므로 조건을 걸어 a<=b라면 a ~ b+1까지의 숫자를 합하고 a>b라면 b ~ a+1까지의 숫자를 합한 결과를 answer에 담아줬습니다.
  • a or b보다 +1을 한 이유는 range(a, b)는 a에서 b-1까지만 return 하는 파이썬의 특성이므로 +1을 더해서 a~b or b~a를 만족하게 함.
profile
Data science

0개의 댓글