알고리즘(C++) / 백준 1292 : 쉽게 푸는 문제

1292

www.acmicpc.net/problem/1292

1292번: 쉽게 푸는 문제

 

문제

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

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

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

 

입력

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

 

출력

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

 

코드

//1292 쉽게 푸는 문제
#include<iostream>
using namespace std;

int main() {
	int A, B;
	int add = 1;
	int num = add; //더해져야할 개수
	int sum = 0;

	cin >> A >> B;

	for (int i = 1; i <= B; i++) {

		if (num == 0) {
			add++;
			num = add;
		}

		if (i >= A)
			sum += add;

		num--;
	}

	cout << sum << endl;
	return 0;
}

 

결과

 

설명

A, B가 각각 입력 받은 두개의 자연수이고, add는 더해져야하는 수, num이 더해져야하는 개수, sum이 출력되어야할 숫자의 합이다. 

num은 add만큼 더해져야 하고, 반복문을 한번 실행할때마다 1씩 감소한다. 그때 num이 0이면 add를 증가시킨다.