[백준/파이썬] 2355번

백준 2355번

문제

https://www.acmicpc.net/problem/2355

풀이

#시간초과
a, b = map(int, input().split())

result  = 0
for i in range(a,b+1):
    result += i

print(result)

#메모리초과
a, b = map(int, input().split())

result = []
for i in range(a,b+1):
    result.append(i)

print(sum(result))

#답
a, b = map(int, input().split())
result = 0 

if a<=b:
    result = (b*(b+1)//2) - (a*(a+1)//2) + a
else:
    result = (a*(a+1)//2) - (b*(b+1)//2) + b

print(result)

풀이

가장 간단한 방법으로 풀었을 때는 시간초과가 떴습니다. 이후, result에 값을 저장하여 sum함수로 풀었을 때는 메모리초과가 떴습니다.
그래서 수학공식을 이용하여, 다시 풀었습니다.