프로그래머스

프로그래머스: 두 정수 사이의 합 | Python

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

 

1. 풀이

def solution(a, b):
    list=[]
    answer = 0
    
    if a<b:
        pass
    else:
        a, b = b, a  #Swap을 간단하게 처리
    
    for i in range(a, b + 1):
        answer += i
    
    return answer

 

 

어렵게 했던 코드 간단하게 처리

#Swap을 간단하게 처리
'''
temp = a
a = b
b = temp
'''
a, b = b, a

 

 

2. 다른사람 풀이

def adder(a, b):
    if a > b:
        a, b = b, a
    return sum(range(a, b + 1))

 

 

sum함수 사용법

sum 함수는 반복 가능한(iterable) 객체를 받아 그 안에 있는 모든 요소의 합을 계산합니다.

 sum(iterable, start=0

  • iterable: 합을 계산할 요소들이 들어있는 반복 가능한(iterable) 객체입니다. 주로 숫자로 이루어진 리스트, 튜플, 집합 등
  • start (선택적): 합산의 초기값으로, 기본값은 0입니다. 이 매개변수를 사용하면 합산에 다른 값으로 시작할 수 있습니다.
# sum(iterable) 리스트
result1 = sum([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
print(result1)  # 출력 : 55


# sum(iterable) 튜플
b = (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
result3 = sum(b)
print(result3)  # 출력 : 55

# sum(iterable, start = 0) 두번째 인자의 값을 정해줬을 때
result3 = sum([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], 1111)
print(result3)  # 출력 : 55 (리스트) + 1111 (start) = 1166