728x90
반응형
문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12912 |
두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수를 작성해야 한다.
a와 b가 같은 경우는 둘 중 아무 수나 리턴하는 조건이 주어져있고, a>b일 수도, a<b일 수도 있다.
따라서 if-else문으로 두 경우를 나누어 작성했다.
def solution(a, b):
answer = 0
if a<=b:
for i in range(a,b+1): #두 정수 사이가 오름차순일 때
answer+=i
else:
for i in range(a,b-1,-1): #두 정수 사이가 내림차순일 때
answer+=i
return answer
range 사용법 보러가기 -> 2020/05/06 - [개발이야기/python] - range 사용하기
728x90
반응형
'알고리즘이야기 > 프로그래머스_알고리즘_lv1' 카테고리의 다른 글
나누어 떨어지는 숫자 배열 (0) | 2020.05.11 |
---|---|
크레인 인형뽑기 (2019 카카오 겨울 인턴십 기출) (0) | 2020.05.11 |
같은 숫자는 싫어 (0) | 2020.05.05 |
가운데 글자 가져오기 (0) | 2020.05.05 |
K번째수 (0) | 2020.05.03 |
댓글