⌨️ Code Taka/Python
[프로그래머스] 두 정수 사이의 합
ny:D
2024. 5. 30. 10:00
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
문제
두 정수 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 |
🍀 Solution
def solution(a, b):
answer = 0
if a> b :
for i in range(b,a+1):
answer+= i
elif a==b:
answer= a
else:
for i in range(a,b+1):
answer += i
return answer
🐧 풀이
- a와 b중 큰 수를 판별하는 if문 작성
- a>b인 경우 b~a까지의 범위에서 하나씩 수를 뽑는 반복문 작성, answer에 i를 더해 b와 a사이에 속한 모든 정수의 합을 계산
- a와 b가 같은 경우 둘 중 아무수나 리턴해야 하므로, a를 리턴
- a<b인 경우에는 a~b까지의 범위에서 하나씩 수를 뽑고 a>b인 경우와 같은 방식으로 answer를 계산