[Python] Programmers/Level1
[프로그래머스/Level1] 두 정수 사이의 합
파크영
2021. 6. 4. 17:53
문제 설명
두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.
제한 조건
- a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
- a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
- a와 b의 대소 관계는 정해져 있지 않습니다.
입출력 예
a | b | return |
3 | 5 | 12 |
3 | 3 | 3 |
5 | 3 | 12 |
나의 풀이
[Python(파이썬)]
def solution(a, b):
answer = 0
if a <= b :
for i in range(0,b-a+1):
answer += (a + i)
else :
for i in range(0,a-b+1):
answer += (b + i)
return answer
알게 된 점
sum(), min(), max()를 사용하니 간결한 코드가 나옴
sum(range(min(a,b),max(a,b)+1))
코딩테스트 연습 - 두 정수 사이의 합
두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요. 예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다. 제한 조건 a와 b가 같은 경우
programmers.co.kr