<백준> 1292번 파이썬 알고리즘

a, b = map(int,input().split())
res = 0
arr = []
arr.append(0)
for i in range(1000):
  for j in range(i):
    if len(arr) < 1001:
      arr.append(i)
for i in range(b-a+1):
  res += arr[a]
  a += 1
print(res)

 

먼저 리스트에 값들을 모두 넣었다.

문제에서 최대값이 1000이므로 1000까지만 리스트 설정했다.