[XSY] 选举（线段树优化dp）

选举

//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;
}