[백준 10025] 게으른 백곰 C++

문제

백준 10025 게으른 백곰 C++

10025번: 게으른 백곰

풀이

일반적인 두 포인터 문제다.

처음부터 마지막까지, 양동이의 양을 더하고 빼주면서 최대합을 구한다.

범위 계산이 귀찮았는데, 정확하게 계산해 범위 초과 이슈가 나지 않도록 유의한다.

소스 코드

#include <iostream>
#include <algorithm>
using namespace std;
 
const int MAX_COUNT = 1000001;
int ice[MAX_COUNT];
 
int main()
{
    int N = 0;
    int K = 0;
 
    cin >> N >> K;
    int maxIdx = 0;
 
    while (N--)
    {
        int value = 0;
        int idx = 0;
        
        cin >> value >> idx;
        ice[idx] = value;
        if (idx > maxIdx)
            maxIdx = idx;
    }
 
    int sum = 0;
 
    for (int idx = 0; idx <= 2 * K; idx++)
    {
        if (idx >= MAX_COUNT)
            break;
 
        sum += ice[idx];
    }
 
    int result = sum;
    int start = 0;
 
    for (int end = 2 * K + 1; end <= maxIdx; end++)
    {
        sum += ice[end];
        sum -= ice[start];
        result = max(result, sum);
        start++;
    }
 
    cout << result;
    return 0;
}