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

2022. 3. 26. 00:30

728x90

출처 : 프로그래머스

코딩테스트 연습 - 두 정수 사이의 합

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요. 예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다. 제한 조건 a와 b가 같은 경우

programmers.co.kr

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

def solution(a, b):
    answer = (abs(a-b)+1)*(a+b)//2
    return answer

수열에서 합 공식을 사용하면 된다.

n(a+l)//2 -> n 은 항 개수, a는 첫 항, l은 마지막 항

say repository