bzoj2962: 序列操作

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

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