⌨️ 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를 계산

🖨️ 결과