첫 코드
def solution(a, b):
x = []
if (a < b):
for i in range(a, b+1):
x.append(i)
answer = sum(x)
elif (b < a):
for i in range(b, a+1):
x.append(i)
answer = sum(x)
else:
answer = a
return answer
개선된 코드
def solution(a, b):
answer = 0
if a > b:
answer = sum(range(b, a + 1))
elif b > a:
answer = sum(range(a, b + 1))
elif a == b:
return a
return answer
처음에는 불필요하게 for문을 추가하였다. 복잡도 면에서 간결한 알고리즘으로 개선하였다.
'알고리즘 > 프래그래머스 lv.1' 카테고리의 다른 글
[프래그래머스] 평균 구하기(Python) (0) | 2022.10.18 |
---|---|
[프래그래머스] x만큼 간격이 있는 n개의 숫자(Python) (0) | 2022.10.18 |
[프래그래머스] 가운데 글자 가져오기(Python) (0) | 2022.10.18 |
[프래그래머스] 서울에서 김서방 찾기(Python) (0) | 2022.10.18 |
[프래그래머스] 수박수박수박수?(Python) (0) | 2022.10.18 |