Description
给定一个长度为n的序列,你有一次机会选中一段连续的长度不超过d的区间,将里面所有数字全部修改为0。
请找到最长的一段连续区间,使得该区间内所有数字之和不超过p。
Input
第一行包含三个整数n,p,d(1<=d<=n<=2000000,0<=p<=10^16)。
第二行包含n个正整数,依次表示序列中每个数wi。
Output
包含一行一个正整数,即修改后能找到的最长的符合条件的区间的长度。
Sample Input
9 7 2
3 4 1 9 4 1 7 1 3
Sample Output
5
HINT
将第4个和第5个数修改为0,然后可以选出区间[2,6],总和为4+1+0+0+1=6。
分析
首先假设我们已经有了一个右端点,如果要使用修改的话,我们肯定是想让左端点尽量的靠前,那么这就要求我们要让这次修改的收益最大化就是选个最大的区间将它的权值修改为0 这样等下可以利用这些松弛尽量左边的点。具体说就是如果当前这个区间是[l,r],那么我们要尽量选择右端点范围是[l,r]的长度为d的区间使它的值最大(一些边界问题自己考虑)然后就差不多解决了。
我分析了半天//一直以为这个是n^2的///尴尬
代码
#include <algorithm>
#include <iostream>
#include <cstring>
#include <complex>
#include <cstdio>
#include <queue>
#include <cmath>
#include <map>
#include <set>
#define N 2200000
#define INF 0x7fffffff
#define ll long long
#define sqr(x) ((x) * (x))
#define pi acos(-1)
ll n,d,p;
ll a[N];
ll sum[N], mx[N];
int q[N];
ll read()
{
ll x = 0, f = 1;
char ch = getchar();
while (ch < '0' || ch > '9'){if (ch == '-') f = -1; ch = getchar();}
while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
return x * f;
}
int main()
{
n = read(), p =read(), d = read();
for (int i = 1; i <= n; i++)
{
a[i] = read();
sum[i] = sum[i - 1] + a[i];
}
for (int i = 1; i <= n; i++)
{
int l = i, r = std::min(l + d - 1, n);
mx[i] = sum[r] - sum[l - 1];
}
int head = 1, tail = 0;
q[++tail] = 1;
int ans = 1;
int len = d;
for (int i = d + 1; i <= n; i++)
{
int now = i - d + 1;
while (head <= tail && mx[now] >= mx[q[tail]]) tail--;
q[++tail] = now;
while (sum[i] - sum[ans - 1] > p + mx[q[head]])
{
ans++;
while (q[head] < ans) head++;
}
len = std::max(len, i - ans + 1);
}
printf("%d\n",len);
}