[백준] 12847번 - 꿀 아르바이트 [Java][C++]
[백준] 12847번 - 꿀 아르바이트 [Java][C++]
1. 문제 풀이
주어진 임금으로 만들 수 있는 수열에서 구간의 길이가 $m$ 인 구간 중 구간 합이 가장 큰 경우를 찾는 문제다. 누적합을 활용해도 해결할 수 있고 슬라이딩 윈도우를 활용해도 해결할 수 있다. 주어진 임금의 합이 정수 타입 범위를 넘어갈 수 있음에 주의해야 한다.
2. 코드
1. 누적합 [Java]
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
import java.io.*;
import java.util.*;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int N = Integer.parseInt(st.nextToken());
int M = Integer.parseInt(st.nextToken());
int[] arr = new int[N];
st = new StringTokenizer(br.readLine());
for (int i = 0; i < N; i++) {
arr[i] = Integer.parseInt(st.nextToken());
}
long[] pSum = new long[1 + N];
for (int i = 1; i <= N; i++) {
pSum[i] = pSum[i - 1] + arr[i - 1];
}
long max = 0;
for (int i = 0; i <= N - M; i++) {
max = Math.max(max, pSum[i + M] - pSum[i]);
}
System.out.println(max);
}
}
2. 슬라이딩 윈도우 [Java]
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
import java.io.*;
import java.util.*;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int N = Integer.parseInt(st.nextToken());
int M = Integer.parseInt(st.nextToken());
int[] arr = new int[N];
st = new StringTokenizer(br.readLine());
for (int i = 0; i < N; i++) {
arr[i] = Integer.parseInt(st.nextToken());
}
// 초기 윈도우의 합 계산
long sum = 0;
for (int i = 0; i < M; i++) {
sum += arr[i];
}
// 윈도우를 이동시키며 최대 이익 계산
long max = sum;
for (int i = 0; i < N - M; i++) {
sum = sum - arr[i] + arr[i + M];
max = Math.max(max, sum);
}
System.out.println(max);
}
}
3. 누적합 [C++]
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
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vector<int> v(n);
for (int& x : v) cin >> x;
vector<long long> psum(1 + n);
for (int i = 1; i <= n; i++) {
psum[i] = psum[i - 1] + v[i - 1];
}
long long mx = 0;
for (int i = 0; i <= n - m; i++) {
mx = max(mx, psum[i + m] - psum[i]);
}
cout << mx;
}
4. 슬라이딩 윈도우 [C++]
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
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vector<int> v(n);
for (int& x : v) cin >> x;
// 초기 윈도우의 합 계산
long long sum = 0;
for (int i = 0; i < m; i++) {
sum += v[i];
}
// 윈도우를 이동하며 최댓값 반복 갱신
long long mx = sum;
for (int i = 0; i < n - m; i++) {
sum = sum - v[i] + v[i + m];
mx = max(mx, sum);
}
cout << mx;
}
3. 풀이 정보
1. 누적합 [Java]
| 언어 | 시간 | 메모리 | 코드 길이 |
|---|---|---|---|
| Java 11 | 276 ms | 25776 KB | 867 B |
2. 슬라이딩 윈도우 [Java]
| 언어 | 시간 | 메모리 | 코드 길이 |
|---|---|---|---|
| Java 11 | 272 ms | 24856 KB | 959 B |
3. 누적합 [C++]
| 언어 | 시간 | 메모리 | 코드 길이 |
|---|---|---|---|
| C++ 17 | 12 ms | 3196 KB | 454 B |
4. 슬라이딩 윈도우 [C++]
| 언어 | 시간 | 메모리 | 코드 길이 |
|---|---|---|---|
| C++ 17 | 8 ms | 2412 KB | 535 B |
This post is licensed under CC BY 4.0 by the author.