https://www.acmicpc.net/problem/1292
문제
동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.
이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다.
하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.
입력
첫째 줄에 구간의 시작과 끝을 나타내는 정수 A, B(1 ≤ A ≤ B ≤ 1,000)가 주어진다. 즉, 수열에서 A번째 숫자부터 B번째 숫자까지 합을 구하면 된다.
출력
첫 줄에 구간에 속하는 숫자의 합을 출력한다.
풀이
vector에 1, 2, 2, 3, 3, 3, 4... 에 해당하는 숫자를 미리 넣어두고
만들어진 arr를 돌면서 범위에 해당하는 숫자를 결과값에 더해준다
코드
#include<iostream>
#include<vector>
using namespace std;
int main()
{
int A, B, cnt = 0, num = 1;
cin >> A >> B;
vector<int>arr;
// 배열이 1000개의 입력값을 다 받을 필요가 없으므로 입력같은 B 값으로 범위를 제한해준다
while (cnt < B)
{
// 현재 num 만큼 돌며 벡터에 추가해준다
for (int i = 0; i < num;i++)
{
arr.push_back(num);
// 만약 B 보다 벡터에 입력받은 수가 많아지면 while문을 탈출한다
cnt++;
if (cnt > B) break;
}
num++;
}
int ans = 0;
// 인덱스는 0부터 시작하기 때문에 A번째 값을 받아오기 위해서는 -1를 빼준다
for (int i = A-1; i < B; i++)
{
ans += arr[i];
}
cout << ans;
}
'study > 백준' 카테고리의 다른 글
[백준] 10178. 할로윈의 사탕 : C++ (0) | 2022.02.13 |
---|---|
[백준] 1009. 분산처리 : C++ (0) | 2022.02.13 |
[백준] 2506. 점수계산 : C++ (0) | 2022.02.10 |
[백준] 2501. 약수 구하기 : C++ (0) | 2022.02.09 |
[백준] 2693. N번째 큰 수 : C++ (0) | 2022.02.05 |