[BAEKJOON] 1292 쉽게 푸는 문제

문제

동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.

이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다.

하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.

입력

첫째 줄에 구간의 시작과 끝을 나타내는 정수 A, B(1 ≤ A ≤ B ≤ 1,000)가 주어진다. 즉, 수열에서 A번째 숫자부터 B번째 숫자까지 합을 구하면 된다.

출력

첫 줄에 구간에 속하는 숫자의 합을 출력한다.

예제 입력 1 복사

3 7

예제 출력 1 복사

15

---

문제를 잘 못 읽고 최적화를 생각해나가다가 

 

import sys

s = 0

n,m = map(int, sys.stdin.readline().strip().split())

tn = n

a = [1, 2, 2]

while len(a) < m:

    a.extend([tn for i in range(tn)])

    tn += 1

print(sum(a[n-1:m]))

 

m 까지만 구하면 모든 리스트를 구할 필요는 없을 테니 했는데 이상하게 되지 않았다.

 

근데 보니까 뭐 1000개까지 밖에 안되니 시간 초는 2초니 1초 이상은 할 수 있다는 것이기에 걍 리스트를 다 만들어버렸다.

 

import sys

s = 0

n,m = map(int, sys.stdin.readline().strip().split())

a = []

for i in range(1, 1001):

    a.extend([i for h in range(i)])

print(sum(a[n-1:m]))