bzoj2962: 序列操作

来源：互联网发布：微信封号知乎编辑：程序博客网时间：2024/06/03 07:34

这是一道区间修改的线段树，其中区间加比较难，但手推一下也可以推出来，然而我慢的飞起。。。

#include<iostream>#include<cstdio>#include<cstring>#define P 19940417#define N 160005#define ll long long#define MO(x) ((x)%P)using namespace std;int n,q,a[N],x,y,z,b[21],tg[N],op[N],C[50005][21];int RE(){int x=0,p=1;char c=getchar();while(c<'0'||c>'9') p=(x=='-')?-1:1,c=getchar();while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();return x*p;}struct T{int a[21],y;void clr(){memset(a,0,sizeof a);}void merge(T x,T y){for (int i=1;i<=20;i++){a[i]=(x.a[i]+y.a[i])%P;for (int j=1;j<i;j++)a[i]=((ll)x.a[j]*y.a[i-j]+a[i])%P;}}void plus(int x){int now;for (int i=20;i;i--){now=x;for (int j=i-1;j;j--,now=(ll)now*x%P)a[i]=((ll)now*C[y-j][i-j]%P*a[j]+a[i])%P;a[i]=((ll)C[y][i]*now+a[i])%P;}}void opp(){for (int i=1;i<=20;i+=2)a[i]=P-a[i];}}Ans,s[N];void down(int k,int l,int r){if (op[k]) s[l].opp(),s[r].opp(),op[k]=0,op[l]^=1,op[r]^=1,tg[l]=P-tg[l],tg[r]=P-tg[r];if (tg[k]) s[l].plus(tg[k]),s[r].plus(tg[k]),tg[l]=MO(tg[l]+tg[k]),tg[r]=MO(tg[r]+tg[k]),tg[k]=0;}void build(int k,int l,int r){s[k].y=r-l+1;if (l==r){s[k].a[1]=a[l];return;}int mid=(l+r)>>1,L=k<<1,R=L|1;build(L,l,mid);build(R,mid+1,r);s[k].merge(s[L],s[R]);}void add(int k,int l,int r){if (x<=l&&r<=y){tg[k]=MO(tg[k]+z);s[k].plus(z);return;}int mid=(l+r)>>1,L=k<<1,R=L|1;down(k,L,R);if (x<=mid) add(L,l,mid);if (y>mid) add(R,mid+1,r);s[k].merge(s[L],s[R]);}void opp(int k,int l,int r){if (x<=l&&r<=y){op[k]^=1;tg[k]=P-tg[k];s[k].opp();return;}ll mid=(l+r)>>1,L=k<<1,R=L|1;down(k,L,R);if (x<=mid) opp(L,l,mid);if (y>mid) opp(R,mid+1,r);s[k].merge(s[L],s[R]);}void qry(int k,int l,int r){if (x<=l&&r<=y){Ans.merge(Ans,s[k]);return;}ll mid=(l+r)>>1,L=k<<1,R=L|1;down(k,L,R);if (x<=mid) qry(L,l,mid);if (y>mid) qry(R,mid+1,r);}int main(){n=RE();q=RE();C[0][0]=1;for (int i=1;i<=n;i++){C[i][0]=1;for (int j=1;j<=i&&j<=20;j++)C[i][j]=MO(C[i-1][j]+C[i-1][j-1]);}for (int i=1;i<=n;i++) a[i]=RE();build(1,1,n);while(q--){char c[1];scanf("%s",c);if (c[0]=='I') x=RE(),y=RE(),z=RE()%P,add(1,1,n);else if (c[0]=='R') x=RE(),y=RE(),opp(1,1,n);else x=RE(),y=RE(),z=RE(),Ans.clr(),qry(1,1,n),printf("%d\n",Ans.a[z]<0?Ans.a[z]+P:Ans.a[z]);}return 0;}

0 0