好题分享luogu 5858 「SWTR-03」Golden Sword

好题分享

luogu P5858 「SWTR-03」Golden Sword

\quad 读完题后不难发现，本题就是一个变形的背包，在转移时多枚举一层$k$作为取出物品的个数即可。可是，就因为多了这一层循环就会使得$TLE$情况发生。怎么办呢？？

状态：$dp[i][j]$表示前$i$个物品，容纳$j$个物品，最大坚固值
状态转移方程：$dp[i][j]=ma{x}_{dp[i-1][j+k-1]}+j\ast a[i](k\in [0,s])$

\quad 这时，让我们分析一下，每一个$dp[i][j]$都是从$dp[i-1]$中某一段连续的值中取最大而转移来的，而且那一段连续的值的长度都为s+1(0~s)，那我们就不难想到可以用滑动窗口维护区间最大值，另开数组记录。
$Code:$

#include<bits/stdc++.h>
using namespace std;
#define int long long

const int maxn = 5e3 + 5;
int n, w, s;
int a[maxn];
int dp[2][maxn * 2];//滚动数组（注意第二维的范围）
int v[2][maxn * 2];//记录从i开始，往后延伸s个位置中，dp[i][j]的最大值
deque < int > qq;

signed main() {
	scanf("%lld%lld%lld", &n, &w, &s);
	for(int i = 1; i <= n; i ++) scanf("%lld", &a[i]);
	memset(dp, 0xcf, sizeof dp);
	for(int i = 0; i <= w; i ++) dp[0][i] = 0;
	for(int i = 1; i <= n; i ++) {
		memset(dp[i & 1], 0xcf, sizeof dp[i & 1]);
		for(int j = 1; j <= min(i, w); j ++) {
			dp[i & 1][j] = max(dp[i & 1][j], v[(i - 1) & 1][j - 1] + 1LL * j * a[i]);
		}
		while(!qq.empty()) qq.pop_back();
		for(int j = 1; j <= w + s - 1; j ++) {//滑动窗口求最大值
			while(!qq.empty() && qq.front() < j - s) qq.pop_front();
			while(!qq.empty() && dp[i & 1][qq.back()] < dp[i & 1][j]) qq.pop_back();
			qq.push_back(j);
			if(j >= s) v[i & 1][j - s] = dp[i & 1][qq.front()];
		}
	}
	int maxx = -LONG_LONG_MAX;
	for(int i = 1; i <= w; i ++)
		maxx = max(maxx, dp[n & 1][i]);
	printf("%lld", maxx);
	return 0;
}