백준 1292 쉽게 푸는 문제 [c++]

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int a, b;	//첫째줄에 시작과 끝을 나타내는 변수
	int idx = 2;	//인덱스로 사용할 변수
	int sum = 0;	//합계로 사용할 변수
	cin >> a >> b;	//a,b 입력받기

	int arr[1001] = { 0, };	//수열을 저장할 배열

	for (int i = 1; i < 1001; i++)	//1부터 1000까지 반복
	{	
		if (i == 1)	//1일 경우에는 한개만 저장하면 됨으로 저장
			arr[i] = i;
		else //그외의 경우 2부터
		{
			for (int j = 0; j < i; j++)	//i의수 만큼 반복
			{
				if (idx > 1000)	//수열은 1000까지 이기때문에 1000을 넘어가면 종료
					break;
				arr[idx] = i;	//배열에 값을 저장
				idx++;	//값만큼 추가해야하기때문에 인덱스만 증가
			}
		}
	}
	
	for (int i = a; i <= b; i++)	
		sum += arr[i];

	cout << sum << '\n';
	return 0;
}