[백준] 2559번 - 수열 [Java][C++]

문제 링크

---

1. 문제 풀이

주어진 날짜들에 대해 연속적인 $K$ 일의 온도의 합 중 가장 큰 값을 구해야 하는 문제다. $N$ 이 최대 $100,000$ 이어서 $O(N^2)$ 풀이로는 해결할 수 없다. 구간의 합을 빠르게 계산할 수 있는 누적합을 활용하거나 슬라이딩 윈도우를 활용하면 효율적으로 해결할 수 있다.

---

2. 코드

1. 누적합 [Java]

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 K = 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());
        }

        int[] pSum = new int[1 + N];
        for (int i = 1; i <= N; i++) {
            pSum[i] = pSum[i - 1] + arr[i - 1];
        }

        // 누적합의 차로 구간합 계산 및 최댓값 갱신
        int max = Integer.MIN_VALUE;
        for (int i = 0; i <= N - K; i++) {
            max = Math.max(max, pSum[i + K] - pSum[i]);
        }

        System.out.println(max);
    }
}

2. 슬라이딩 윈도우 [Java]

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 K = 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());
        }

        // 초기 윈도우의 합 계산
        int sum = 0;
        for (int i = 0; i < K; i++) {
            sum += arr[i];
        }

        // 윈도우를 이동하며 최댓값 반복 갱신
        int max = sum;
        for (int i = 0; i < N - K; i++) {
            sum = sum - arr[i] + arr[i + K];
            max = Math.max(max, sum);
        }

        System.out.println(max);
    }
}

3. 누적합 [C++]

#include <bits/stdc++.h>
using namespace std;

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

    int n, k;
    cin >> n >> k;

    vector<int> v(n);
    for (int& x : v) cin >> x;

    vector<int> psum(1 + n);
    for (int i = 1; i <= n; i++) {
        psum[i] = psum[i - 1] + v[i - 1];
    }

    // 누적합의 차로 구간합 계산 및 최댓값 갱신
    int mx = INT_MIN;
    for (int i = 0; i <= n - k; i++) {
        mx = max(mx, psum[i + k] - psum[i]);
    }

    cout << mx;
}

4. 슬라이딩 윈도우 [C++]

#include <bits/stdc++.h>
using namespace std;

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

    int n, k;
    cin >> n >> k;

    vector<int> v(n);
    for (int& x : v) cin >> x;

    // 초기 윈도우의 합 계산
    int sum = 0;
    for (int i = 0; i < k; i++) {
        sum += v[i];
    }

    // 윈도우를 이동하며 최댓값 반복 갱신
    int mx = sum;
    for (int i = 0; i < n - k; i++) {
        sum = sum - v[i] + v[i + k];
        mx = max(mx, sum);
    }

    cout << mx;
}

---

3. 풀이 정보

1. 누적합 [Java]

언어	시간	메모리	코드 길이
Java 11	236 ms	23952 KB	949 B

2. 슬라이딩 윈도우 [Java]

언어	시간	메모리	코드 길이
Java 11	244 ms	23988 KB	957 B

3. 누적합 [C++]

언어	시간	메모리	코드 길이
C++ 17	8 ms	2804 KB	513 B

4. 슬라이딩 윈도우 [C++]

언어	시간	메모리	코드 길이
C++ 17	8 ms	2412 KB	523 B

---