洛谷__P12882 [蓝桥杯 2025 国 C] 数列染色（单调队列+DP）

题目链接：P12882 [蓝桥杯 2025 国 C] 数列染色 - 洛谷

---

题目大意：

长度为 n 的数列，初始时只有首尾为黑色其余均为白色。

每次可将白色数染黑，要求相邻黑数之间不超过 k 个白数。

求染色后黑数之和的最小值。

---

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<unordered_set>
#include<unordered_map>
#include<bitset>
#include<tuple>
#include<array>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/hash_policy.hpp>
#include <ext/numeric>
#define inf 72340172838076673
#define int long long
#define endl '\n'
#define F first
#define S second
#define mst(a,x) memset(a,x,sizeof (a))
#define gmap __gnu_pbds::gp_hash_table
#define power __gnu_cxx::power
using namespace std;
typedef pair<int, int> pii;

const int N = 200086, mod = 998244353;

int n, m;
int a[N];
int f[N];

void solve() {

    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    
    deque<int> q;
    f[1] = a[1];
    q.push_back(1);
    for (int i = 2; i <= n ; i++) {
        while (q.size() && i - q.front() - 1 > m) q.pop_front();
        f[i] = f[q.front()] + a[i];
        while (q.size() && f[i] <= f[q.back()]) q.pop_back();
        q.push_back(i);
    }
    
    cout << f[n] << endl;
    
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    
    int T = 1;
    // cin >> T;
    while (T--) solve();
    
    return 0;
}