BOJ: 1292 쉽게 푸는 문제

문제

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

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

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

입력

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

출력

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

예제 입력 1

3 7

예제 출력 1

15

코드

#include <bits/stdc++.h>

using namespace std;

int main(void) {
    int s[1001];
    int a, b, k=1, i=0;
    int sum = 0;
    bool flag = false;

    while(1) {
        for(int j=0; j<k; j++) {
            if(i == 1000) {
                flag = true;
                break;
            }
            
            s[i++] = k;
        }

        if(flag == true)
            break;

        k++;
    }
        
    cin>>a>>b;

    for(int i=a-1; i<b; i++)
        sum += s[i];

    cout<<sum;

    return 0;
}

Reference

BOJ