//a[i]=1 or 0 or -1,表示i支持小奇 or 中立 or 支持魔法猪
//sum[i]=a[1]+a[2]+...+a[i]
//设dp[i]表示把1~i划分成若干区间,(小奇得票-魔法猪得票)的最大值
/*dp[i]=max {
dp[j]+1 (sum[i]-sum[j]>0即sum[i]>sum[j])
dp[j] (sum[i]-sum[j]=0即sum[i]=sum[j])
dp[j]-1 (sum[i]-sum[j]<0即sum[i]<sum[j])
}(max(0,i-R)<=j<=max(0,i-L))*/
//int get(int x){if(x>0)return 1;if(x<0)return -1;if(x==0) return 0;}
//对于某一段区间中的j,dp[j]+get(sum[i]-sum[j])的最大值是很难维护的,但dp[j]的最大值很好维护
/*所以考虑这么求dp[i]:dp[i]=max{
Max{dp[j]}(sum[j]<sum[i]) +1
Max{dp[j]}(sum[j]=sum[i])
Max{dp[j]}(sum[j]>sum[i]) -1
}(max(0,i-R)<=j<=max(0,i-L))*/
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int N=1e6+10;
const int inf=1e9;
int read(){
int 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 n,L,R,a[N],sum[N],dp[N];
vector<int> v[N<<1];
int l[N],r[N],pos[N],tot;
struct SegmentTree{
int mx[N<<2];
void build(int u,int l,int r){
mx[u]=-inf;
if(l==r) return;
int mid=(l+r)>>1;
build(u<<1,l,mid);
build(u<<1|1,mid+1,r);
}
void modify(int u,int l,int r,int pos,int val){
if(l==r){
mx[u]=val;
return;
}
int mid=(l+r)>>1;
if(pos<=mid) modify(u<<1,l,mid,pos,val);
else modify(u<<1|1,mid+1,r,pos,val);
mx[u]=max(mx[u<<1],mx[u<<1|1]);
}
int query(int u,int l,int r,int L,int R){
if(L>R) return -inf;
if(L<=l&&r<=R) return mx[u];
int mid=(l+r)>>1;
if(R<=mid) return query(u<<1,l,mid,L,R);
else if(L>mid) return query(u<<1|1,mid+1,r,L,R);
else return max(query(u<<1,l,mid,L,mid),query(u<<1|1,mid+1,r,mid+1,R));
}
}T;
void ins(int i){
T.modify(1,0,n,pos[i],dp[i]);
}
void del(int i){
T.modify(1,0,n,pos[i],-inf);
}
int main(){
n=read();L=read();R=read();
for(int i=1;i<=n;i++) a[i]=read();
for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
for(int i=0;i<=n;i++) v[sum[i]+n].push_back(i);
for(int i=0;i<=n+n;i++){
int lst=tot;
for(int j=0;j<v[i].size();j++) pos[v[i][j]]=tot++;
for(int j=0;j<v[i].size();j++) l[v[i][j]]=lst,r[v[i][j]]=tot-1;
}
dp[0]=0;
T.build(1,0,n);
for(int i=1;i<=n;i++){
if(i-R-1>=0) del(i-R-1);
if(i-L>=0) ins(i-L);
int ta=T.query(1,0,n,0,l[i]-1);
int tb=T.query(1,0,n,r[i]+1,n);
int tc=T.query(1,0,n,l[i],r[i]);
if(ta!=-inf) ta++;
if(tb!=-inf) tb--;
dp[i]=max(ta,max(tb,tc));
}
if(dp[n]==-inf) printf("Impossible\n");
else printf("%d\n",dp[n]);
return 0;
}
[XSY] 选举(线段树优化dp)
最新推荐文章于 2024-03-20 21:19:38 发布