코딩테스트 기록/02. Array

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

박세류 2023. 10. 22. 14:16
 

1292번: 쉽게 푸는 문제

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

www.acmicpc.net

배열 기본문제

생각나는대로 풀다보니까 코드가 길어진 감이 있다.

문제의 요구사항에 맞게 주욱 늘려쓴 느낌이다.

A, B = map(int, input().split())
su = list()
cnt = 0
i = 1
total = 0
su.append(0)
for idx in range(1001):
    su.append(i)
    cnt += 1
    if cnt == i:
        cnt = 0
        i += 1

for i in range(A, B + 1):
    total += su[i]

print(total)

 

728x90