[프로그래머스 lv 1] 두 정수 사이의 합.py

<문제 설명>

https://programmers.co.kr/learn/courses/30/lessons/12912

코딩테스트 연습 - 두 정수 사이의 합

두 정수가 주어지면, 그 정수 사이에 있는 모든 정수의 합을 반환하면 되는 문제이다.

 

입출력의 예는 다음과 같다.

a	b	return
3	5	12
3	3	3
5	3	12

 

<문제 해결 아이디어>

반복문으로 해결 가능한 문제이다.

다만 주의할 것은 항상 a가 b보다 작거나 같지 않다는 점이다.

따라서 이 부분을 조건문으로 제한을 두어, 항상 a가 b보다 작거나 같도록 설정하였다.

 

<문제 풀이 코드>

def solution(a, b):
    answer = 0
    if a>b:
        a,b = b,a
    for i in range(a,b+1):
        answer += i
    return answer

채점 결과