[백준 1292] 쉽게 푸는 문제

문제 (Silver 5)

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

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

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

입력

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

출력

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

문제 풀이

• 반복문을 통해 1번부터 B번까지 누적합을 구한 후 Array에 저장
• A ~ B 의 구간합 = (1 ~ B 의 누적합) - (1 ~ (A-1)) 의 누적합

코드

#include <iostream>

using namespace std;

int arr[1050];

int main() {
    int A, B;
    cin >> A >> B;
    
    int cnt = 0;
    int num = 1;
    while(cnt <= B) {
        for(int i=1; i<=num; ++i) {
            arr[cnt+i] = arr[cnt+i-1] + num;
        }
        cnt += num;
        num++;
    }

    cout << arr[B] - arr[A-1];
    
    return 0;
}

문제 링크

1292번: 쉽게 푸는 문제