[알고리즘] 두 정수 사이의 합

유동헌·2021년 9월 14일
0

1일1알고리즘

목록 보기
8/12

링크

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

풀이

def solution(a, b):
    
    answer = 0
    
    for i in range(a, b+1):
        answer += i
    
    if a > b:
        for i in range(b, a+1):
            answer += i
    
    return answer
        
        
print(solution(3, 5))
print(solution(3, 3))
print(solution(5, 3))
  • 설명할 것도 없는 코드.. 빨리 풀고 간단하게 풀기는 했지만, 부끄러워지는 코드다.
  • 절대값, min/max 등을 생각하다가 그냥 for loop로 먼저 해볼까하고 풀어버렸다.

다른 풀이

def solution(a, b):
    
    return sum(range(min(a,b), max(a,b)+1))

print(solution(3, 5))
print(solution(3, 3))
print(solution(5, 3))
  • 큰 값, 작은 값 골라서 리턴해주는 방식
profile
지뢰찾기 개발자

0개의 댓글