[구현/수학] 백준 1292 쉽게 푸는 문제 - Python

 

[ Contents ]

 

 

1. 문제 (링크 참조)

1292번: 쉽게 푸는 문제

 

 

 

2. 문제 풀이

1 2 2 3 3 3 4 4 4 4 ....

 위와 같은 규칙을 가진 수열의 구간합을 구하는 문제입니다.

 

2022.01.19 - [Algorithm] - [Algorithm] 단골 1번 문제, 구현 / 수학

[Algorithm] 단골 1번 문제, 구현 / 수학

 먼저 수열의 규칙대로 구현하고, 구간의 합을 구하면 됩니다.

 

 

 

3. 코드

A, B = map(int, input().split())

cnt = 0
res = 0
for i in range(1000):
    for j in range(i):
        cnt += 1
        if cnt >= A:
            res += i
        if cnt > B-1:
            break
    if cnt > B-1:
        break
print(res)

 1, 2, 3... k 순으로 1씩 커지며, 그 숫자만큼 반복합니다. 바깥 for문에서는 1씩 증가시키고, 안쪽 for문에서는 k만큼 반복해서 수열의 패턴을 구현합니다.

 수열을 구현한 뒤, cnt(count)을 셉니다. A 이상일 때부터 합을 구하고, B를 넘어가면 반복문을 빠져나옵니다.