백준 문제/문자열,누적합,구현
1292: 쉽게 푸는 문제
goldpig
2024. 3. 28. 18:19
728x90
https://www.acmicpc.net/problem/1292
1292번: 쉽게 푸는 문제
첫째 줄에 구간의 시작과 끝을 나타내는 정수 A, B(1 ≤ A ≤ B ≤ 1,000)가 주어진다. 즉, 수열에서 A번째 숫자부터 B번째 숫자까지 합을 구하면 된다.
www.acmicpc.net
설명
동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.
이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다. 하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.
핵심 아이디어
1 22 333 4444를 담을 수 있는 배열을 선언한다. 이때, 1부터 시작하기 위해 배열에 0을 미리 초기화 놓는다. 2중 반복문을 사용해서 배열에 숫자에 해당하는 개수만큼 배열에 저장해준다.
- data = [0] 이렇게 0을 미리 넣어주면 index를 1부터 편하게 계산할 수 있다.
코드 구현
Python
a, b = map(int, input().split())
data = [0]
sum = 0
for i in range(1, b+1):
for j in range(i): # 숫자에 해당하는 만큼 반복해서 더한다.
data.append(i)
for i in range(a, b+1):
sum += data[i]
print(sum)
C++
#include <iostream>
#include <vector>
using namespace std;
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int a, b;
cin >> a >> b;
// 동적 배열 생성
vector<int> data;
data.push_back(0); // data[0] = 0으로 초기화
int sum = 0;
// 해당 숫자 만큼 반목해서 더한다.
for (int i = 1; i <= b; i++) {
for (int j = 0; j < i; j++) {
data.push_back(i);
}
}
// 구간에 따른 결과 출력
for (int i = a; i <= b; i++) {
sum += data[i];
}
cout << sum;
}
728x90