코테01) 두정수사이합

📝문제 설명

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

처음 시도:

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

i) 3개의 케이스를 생각함
ii) 사이에 있는 값들은 for loop를 사용해서 더하는 걸로 선택함
-> 다른 사람들 풀이에 비해 내 풀이가 길다고 생각함

✔ Tips to improve:

sum, min, max function을 사용해보기

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

range(a,b)를 사용해서 sum function에 들어갈 수들을 정해주기
+) a는 포함하나 b는 포함하지 않기 때문에 +1 해주기

이런 함수들이 있다는 것을 알고 있었음에도 사용할 때는 까먹는다
연습을 통해 익숙해져야겠다
한줄의 코드로 쉽게 풀리는 문제였다니..ㅠ