(C++) - 백준(BOJ) 10025번 : 게으른백곰 답

https://www.acmicpc.net/problem/10025

10025번: 게으른 백곰

간단한 부분합 dp였습니다.

 

#include <iostream>
#include <vector>
#include <algorithm>
#define ll long long
using namespace std;
ll ice[1000002];
ll d[1000002];
int main() {
    //양동이 개수, 영옆k, 얼음개수,양동이 좌표
    ll n, k, g, x;
    ll ans = 0;
    cin >> n >> k;
    for (int i = 0; i < n; i++)
    {
        cin >> g >> x;
        ice[x] = g;
    }
    d[0] = ice[0];
    for (int i = 1; i <= 1000000; i++)
    {
        d[i] = d[i - 1] + ice[i];
    }
    if (k <= 1000000)
    {
        for (int i = 0; i <= 1000000; i++)
        {
            //현재위치 - k이전애들을 빼줘야함
            if (i + k <= 1000000 && i-k >=1)
                ans = max(ans, d[i + k] - d[i - k - 1]);
        }
        cout << ans << '\n';
    }
    else
        cout << d[1000000] << '\n';
}