题意:嫖了一张题意图。
思路:一开始以为是dp,发现dp个🔨,然后因为dp[i]的取值为max(dp[j]+{sum[i]-sum[j]}),所以应该是个线段树问题。
当x=0时,{x}=0;当x<0时,{x}=-1;当x>0时,{x}=1;
然后就是区间的x+{y}问题。
一开始线段树各种没想法,后来突然想到权值线段树,然后就很明显正解,然后再中途还需要配个单调队列。
因为y是由sum[i]-sum[j]决定的,所以可以在sum[j]放入dp[j],这样sum[i]减去(1,sum[i]-1)中的任何值都是大于0的,即贡献为1,那么找这个区间的最大值然后加上1就是这段区间的最大贡献了,其余两种贡献为0和贡献为-1的同理。
由于是固定的L和R,所以就相当于个滑动窗口,然后在添加和删除值时需要优先队列。
恶心人的是双端队列卡内存,然后就用vector代替了双端队列,双端队列的内存得是vector十倍还多(估测),反正大的挺离谱。
#include<iostream>
#include<cstdio>
#include<deque>
#include<vector>
using namespace std;
const int MAX_N=1000100;
const int INF=0x3f3f3f3f;
int sum[MAX_N];
struct node{
int l,r,max;
}a[MAX_N*8];
void update(int k){
a[k].max=max(a[k<<1].max,a[k<<1|1].max);
}
void build(int k,int l,int r){
a[k].l=l;a[k].r=r;
if(l==r){
a[k].max=-INF;
return;
}
int mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
update(k);
}
void change(int k,int x,int y){
if(a[k].l==a[k].r){
a[k].max=y;
return;
}
int mid=(a[k].l+a[k].r)>>1;
if(x<=mid)
change(k<<1,x,y);
else
change(k<<1|1,x,y);
update(k);
}
int query(int k,int l,int r){
if(a[k].l>=l&&a[k].r<=r)
return a[k].max;
int mid=(a[k].l+a[k].r)>>1;
int x=-INF;
if(r>mid)
x=max(x,query(k<<1|1,l,r));
if(l<=mid)
x=max(x,query(k<<1,l,r));
return x;
}
int dp[MAX_N];
struct skt{
int val,num;
};
//deque<skt>q[2*MAX_N];
vector<skt>q[2*MAX_N];
int be[2*MAX_N],ed[2*MAX_N];
bool Empty(int x){
if(be[x]<=ed[x])
return false;
else
return true;
}
skt Front(int x){
return q[x][be[x]];
}
skt Back(int x){
return q[x][ed[x]];
}
void Pop_front(int x){
be[x]++;
}
void Pop_back(int x){
ed[x]--;
}
void Push_back(int x,skt t){
if(ed[x]+1<q[x].size()){
ed[x]++;
q[x][ed[x]]=t;
}
else{
ed[x]++;
q[x].push_back(t);
}
}
int main(void){
int T,n,i,L,R,x;
cin>>T;
while(T--){
scanf("%d%d%d",&n,&L,&R);
dp[0]=0;
for(i=1;i<=n;i++){
scanf("%d",&x);
sum[i]=sum[i-1]+x;
dp[i]=-INF;
}
int MK=n+10;
int len=n+MK+10;
build(1,1,len);
for(i=1;i<=len;i++){
q[i].clear();
be[i]=0;
ed[i]=-1;
}
for(i=L;i<=n;i++){
skt t;
int val=dp[i-L],num=i-L;
int k=sum[i-L]+MK;
//while(!q[k].empty()){
while(!Empty(k)){
//t=q[k].back();
t=Back(k);
if(t.val>val)
break;
else
Pop_back(k);
//q[k].pop_back();
}
t.val=val,t.num=num;
//q[k].push_back(t);
Push_back(k,t);
//t=q[k].front();
t=Front(k);
if(t.val<=dp[i-L])
change(1,sum[i-L]+MK,dp[i-L]);
x=query(1,1,sum[i]-1+MK);
dp[i]=max(dp[i],x+1);
x=query(1,sum[i]+MK,sum[i]+MK);
dp[i]=max(dp[i],x);
x=query(1,sum[i]+1+MK,len);
dp[i]=max(dp[i],x-1);
k=sum[i-R]+MK;
//if(i-R>=0&&!q[k].empty()){
if(i-R>=0&&!Empty(k)){
//t=q[k].front();
t=Front(k);
if(t.num==i-R){
//q[k].pop_front();
Pop_front(k);
//if(!q[k].empty()){
if(!Empty(k)){
//t=q[k].front();
t=Front(k);
change(1,sum[i-R]+MK,t.val);
}
else
change(1,sum[i-R]+MK,-INF);
}
}
}
printf("%d\n",dp[n]);
}
return 0;
}