백준 1292

문제

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

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

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

입력

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

출력

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

예제 입력 1

3 7

예제 출력 1

15

풀이(kotlin)

private val r = System.`in`.bufferedReader()

fun main() {
    val (A, B) = r.readLine().split(" ").map { it.toInt() }
    val a = IntArray(B+1)

    var idx = 1
    loop@ for (i in 1.. 1000) {
        loop2@ for (j in 1 .. i) {
            if (idx == B+1) break@loop
            a[idx] = a[idx-1] + i
            idx += 1
        }
    }
    println("${a[B] - a[A-1]}")
}

풀이(C99)

#include <stdio.h>

int main() {
    int a[1000];
    int i, j, idx = 0;
    int x, y, sum = 0;
    for (i = 1; idx != 1000; i++)
        for (j = 1; j <= i; j++) {
            a[idx] = i;
            idx++;
            if (idx == 1000) break;
        }
    scanf("%d %d", &x, &y);
    for (i = x - 1; i <= y - 1; i++)
        sum = sum + a[i];
    printf("%d", sum);
    return 0;
}