PROG-1157: ส่งกระแสไฟฟ้า (electricity)_deque

#include <bits/stdc++.h>

using namespace std;

using lli = long long;
const int N = 500010;
using pli = pair <lli, int>;

int main(){

    int n, k;
    scanf("%d%d", &n, &k);

    deque <pli> dq;

    lli ans = 0;
    for(int i=1;i<=n;i++){
        lli x;
        scanf("%lld", &x);

        while(dq.size() > 0 and i - dq.front().second > k) dq.pop_front();

        if(i == 1) ans = x;
        else ans = x + dq.front().first;

        while(dq.size() > 0 and dq.back().first >= ans) dq.pop_back(); // ต้องหารค่าที่น้อยที่สุด จึงลบค่าที่มากกว่าทิ้งไป

        dq.push_back({ans, i});
    }
    printf("%lld", ans);

    return 0;
}