https://www.acmicpc.net/problem/1806
1806번: 부분합
문제 10,000 이하의 자연수로 이루어진 길이 N짜리 수열이 주어진다. 이 수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이를 구하는 프로그램을 작성하시오. 입력 첫째 줄에 N (10 ≤ N < 100,000)과 S (0 < S ≤ 100,000,000)가 주어진다. 둘째 줄에는 수열이 주어진다. 수열의 각 원소는 공백으로 구분되어져 있으며, 10,000이하의 자연수이다. 출력 첫째 줄에 구하고자 하는 최소의 길
www.acmicpc.net
수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이를 구하는 문제이다.
left와 right 변수 두 개를 포인터로 놓고 구해줬다.
left는 부분 수열의 시작 부분, right는 부분 수열의 마지막 부분을 가리킨다.
가장 짧은 것의 길이를 구하기 위해 최대 길이인 n+1을 ans에 저장해놓는다.
그럼 이제 부분합을 s와 비교해가며 구해주면 된다.
먼저 합이 s보다 작은 경우에는 right를 증가시켜서 합에 수열의 원소를 더해준다.
합이 s보다 클 경우에는 left가 가리키는 원소를 합에서 빼준 후에 left값을 뒤로 옮긴다.
이때, s보다 큰 경우도 정답에 포함이므로 그때의 부분 수열의 길이(right-left+1)를 구해준다.
마찬가지로 합이 s가 되었을 때도 부분 수열의 길이를 구해주면 된다.
이 과정을 left가 right보다 커져버리거나 right가 범위를 벗어나는 경우가 될 때까지 반복한다.
합이 s이상이 되는 경우가 없다면 처음에 저장해 놓은 n+1 값이 ans에 저장되어 있으므로
ans가 그대로 n+1 인 경우에는 문제 조건에 따라 0을 출력한다.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#include <iostream>
using namespace std;
int n, s;
int num[100000];
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n >> s;
for (int i = 0; i < n; i++) {
cin >> num[i];
}
int ans = n+1;
int sum = num[0];
int left = 0;
int right = 0;
while(left <= right && right < n) {
if (sum < s) {
right++;
sum += num[right];
}
else if (sum > s) {
int len = right - left + 1;
if (ans > len) ans = len;
sum -= num[left];
left++;
}
else {
int len = right - left + 1;
if (ans > len) ans = len;
right++;
sum += num[right];
}
}
if (ans == n + 1) ans = 0;
cout << ans << '\n';
return 0;
}
Colored by Color Scripter
|
'BOJ' 카테고리의 다른 글
| [BOJ] 17140. 이차원 배열과 연산 (0) | 2019.07.30 |
|---|---|
| [BOJ] 17142. 연구소 3 (0) | 2019.07.30 |
| [BOJ] 2003. 수들의 합 2 (0) | 2019.07.15 |
| [BOJ] 2748. 피보나치 수 2 (0) | 2019.07.15 |
| [BOJ] 14889. 스타트와 링크(비트마스크 풀이) (0) | 2019.07.11 |