Algorithm/1일 1코테

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

대인보우 2020. 9. 20. 23:22
반응형

문제

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

 

입출력 예시

A B RETURN
3 5 12
3 3 3
5 3 12

 

내 답

def solution(a, b):
    c = min(a,b)
    d = max(a,b)
    answer = 0
    
    for i in range(c,d+1):
        answer +=i

    return answer

 

알고리즘

1. a,b 중 작은 값을 c, 큰 값을 d에 저장

2. c에서 d까지 i 출력 --> answer에 차곡차곡 더해줌

 

다른사람 풀이

def adder(a, b):
    # 함수를 완성하세요
    if a > b: a, b = b, a

    return sum(range(a,b+1))

a가 더 작을땐 그대로고 a가 더 큰 경우 a, b = b, a로 바꿔줌 

range(a, b+1)까지 sum!

 

sum 저렇게 할 수 있는지 첨 알음 ㅋㅋ .ㅋㅋ.ㅋㅋ

반응형