题目大意:最大子序和问题,但有一个限制:不能连续取k个及以上的数。
题目思路:
如果是单纯的最大子序和问题可以使用贪心去解决,但有了不能连续取超过k个数的限制就不能简单的使用贪心了,那我们该怎么办呢?
仔细分析一下限制,我们发现要将目光定位在一个具体的位置i上,这样才方便处理不能取连续k个以上数,即我不取i,我单取i不取i-1,我取i和i-1不取i-2等等这样就保证了我不会取到连续的超过k个数了。
定义状态为dp【i】为以i结尾的满足条件的最大子序和,状态转移方程为:
dp[i]=max(dp[j-1])+sum[i]-sum[j] i-k<=j<=i;
优化:看到方程的形式类似dp[i]=max(dp[j]+b[j])+a[i]。考虑使用单调队列进行优化,看一下窗口是否有重叠,当i-》i+1时,dp[j-1]-sum[j]存在很大重叠所以可以使用单调队列来优化。
以下为参考代码:
#include<iostream>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int N = 1e5 + 10;
int n, k;
long long sum[N];
long long e[N];
long long dp[N],ds[N];
//dp[i]=max(dp[j-1])+sum[i]-sum[j]; i-k<=j<=i
deque<int>q;
long long q_max(int j)
{
ds[j] = dp[j - 1] - sum[j];
while (!q.empty() && ds[q.back()] < ds[j])
{
q.pop_back();
}
q.push_back(j);
while (!q.empty() && q.front() < j - k)
{
q.pop_front();
}
return ds[q.front()];
}
int main()
{
cin >> n >> k;
sum[0] = 0;
for (int i = 1; i <= n; i++)
{
cin >> e[i];
sum[i] = e[i] + sum[i - 1];
}
q.push_back(0);//如果不放0会导致1号位置取不到。
for (int i = 1; i <= n; i++)
{
dp[i] = q_max(i) + sum[i];
}
cout << dp[n] << endl;
}