동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.
이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다.
하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.
위 식을 만족하는 수열에서 구간의 합을 구하자.
미리 수열을 만들어 리스트에 저장한 후 파이썬의 슬라이싱을 이용해 구간합을 구하자.
if __name__ == '__main__':
count = 0
num_list = []
while count < 1000:
count += 1
for i in range(count):
num_list.append(count)
num_input = list(map(int, input().split()))
print(sum(num_list[num_input[0] - 1:num_input[1]]))
수열을 만들 수 있다면 쉽게 해결할 수 있는 문제였다.