選舉

//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]) +1Max{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）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。