CodeForces - 1077F2 Pictures with Kittens (单调队列优化 dp )

题目链接

题意：选择 $x$ 个数， 使得在每个长度为 $k$ 的区间中至少有一个数被选中，选择的这些数的和最大

考虑 $dp$ ，$dp[i][j]$ 表示到第 $i$ 个数的位置时，选了 $j$ 个数的最大值
那么转移就很显然了

$dp[i][j]$ $=$ ${\sum }_{l=1}^k$$max(dp[i][j],dp[i-l][j-1]+v[i])$

但是这样复杂度是 $O(nkx)$，显然不行，但这个${\sum }_{l=1}^k$看起来好像很容易优化掉！

有人回答：线段树！！

然而我用 单调队列 ！！

还有一个值得注意的地方，$dp[i][j]$ 要初始化为 $-INF$，$dp[0][0]=0$，原理类似背包。

代码：

#include <bits/stdc++.h>
#include <ext/rope>
using namespace __gnu_cxx;
using namespace std;
#define mst(a,b) memset(a,b,sizeof(a))
#define ALL(x) x.begin(),x.end()
#define pii pair<int,int>
#define eps 1e-6
#define debug(a) cout << #a": " << a << endl;
#define eularMod(a, b) a < b ? a : a % b + b
inline int lowbit(int x){ return x & -x; }
const int N = 1e5 + 10;
const int mod = (int) 1e9 + 7;
const int INF = 0x3f3f3f3f;
const long long LINF = (1LL << 62);
typedef long long LL;
typedef unsigned long long ULL;
const double PI = acos(-1.0);

int n, k, x, v[N];
LL dp[5005][5005];
pair<int, LL> Q[N];

int main() {
#ifdef purple_bro
    freopen("in.txt", "r", stdin);
//    freopen("out.txt","w",stdout);
#endif
    scanf("%d%d%d", &n, &k, &x);

    for (int i = 1; i <= n; i++) {
        scanf("%d", &v[i]);
    }

    mst(dp, -0x3f);
    dp[0][0] = 0;

    for (int j = 1; j <= x; j++) {
        int rear = 0, top = 0;

        for (int i = 0; i < j; i++) {
            for (;top - rear && Q[top - 1].second <= dp[i][j - 1];) top--;
            Q[top++] = make_pair(i, dp[i][j - 1]);
        }

        for (int i = j; i <= n; i++) {
            for (;top - rear && Q[rear].first < i - k;) rear++;
            // 不在 k 区间内的出队

            dp[i][j] = max(dp[i][j], Q[rear].second + v[i]);

            for (;top - rear && Q[top - 1].second <= dp[i][j - 1];) top--;
            // 维护一个 dp 值单调递减的单调队列，值最大的就是队首辣
            
            Q[top++] = make_pair(i, dp[i][j - 1]);
        }
    }

    LL ans = -1;
    for (int i = n - k + 1; i <= n; i++) {
    	// 确保到最后一个区间，每个区间都最少有一个选中
        ans = max(ans, dp[i][x]);
    }

    printf("%lld\n", ans);
    return 0;
}