Python_두 사이의 정수 합 구하기

김보람·2022년 4월 12일
0

Python

목록 보기
1/39

<문제>

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

<풀이>

def solution(a, b):
    intsum = 0
    if a == b:
        return a
    elif a > b:
        for i in range(b, a+1):
            intsum += i        
        return intsum
    else:
        for i in range(a, b+1):
            intsum += i        
        return intsum

<다른 풀이>

방법 1.
def solution(a,b):
    if a < b:
        return sum(list(range(a,b+1)))
    else:
        return sum(list(range(b,a+1)))
방법 2.
def solution(a,b):
    if a < b:
        return sum([i for i in range(a,b+1)])
    else:
        return sum([i for i in range(b,a+1)])
방법 3.
def solution(a,b):
    return (abs(a-b)+1)*(a+b)//2
방법 4.
def solution(a,b):
    return sum(range(min(a,b), max(a,b)+1))

<생각>

절대값은 너무했다

profile
starry_developer

0개의 댓글