[백준] 1292번 쉽게 푸는 문제 (Python)

1292번: 쉽게 푸는 문제

문제

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

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

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

입력

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

출력

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

 

과정

• 합을 구하고 싶은 시작과 끝을 입력받는다.
• 입력받은 끝점까지의 수열을 만든 후 시작점까지의 합을 구하여 출력한다.

 

예제 입력

3 7

예제 출력

15

 

풀이

n,m = map(int, input().split())

lst = []

for i in range(1, m+1):
    for j in range(i):
        lst.append(i)

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

 

한 줄씩 해석해보기

n,m = map(int, input().split())

lst = []

• 합을 구하고 싶은 시작과 끝 점을 입력받는다.
• 수열을 저장할 리스트를 생성한다. 

for i in range(1, m+1):
    for j in range(i):
        lst.append(i)

• 1부터 끝점+1 까지의 for 문을 생성 후 그 안에 각 숫자만큼의 for문을 하나 더 생성해 리스트에 저장한다.
• 즉 i가  1이면 1개 2이면 2개 3이면 3개가 저장된다.

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

• 리스트의 인덱스 n-1부터 m까지의 합을 출력한다.