题目大意:给定n个数,要求支持区间加,区间取相反数,区间查询任意选c(c<=20)个数的所有方案中乘积的和
和维护k次方的和很像,想要维护选c个数,就要把选1~c个数的方案全部维护出来
这样当合并两个区间的时候(pushup),只需要枚举左右区间分别取了几个数即可
现在考虑两种修改操作:
1.区间取相反数
发现这种操作只会影响区间选奇数个数的情况,把这些答案全部变成相反数即可
2.区间加
这个就比较厉害了,首先我们考虑c=2的情况
设总个数为len,这些数分别为
a1,a2...alen
,则
∑i=1len∑j=1i−1(ai+v)(aj+v)=∑i=1len∑j=1i−1(aiaj+v2+(ai+aj)v)=∑i=1len∑j=1i−1aiaj+C2lenv2+C1len−1∑i=1lenaiv
同理可求得c=3时的式子,所以只需要预处理出组合数,就可以递推了
总时间复杂度
O(MlogN∗202)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100010
using namespace std;
struct ppp{int a[21],len;};
int a[N];
int C[N][21];
int mod=19940417;
int l[N<<2],r[N<<2],t[N<<2];
bool rev[N<<2];
ppp w[N<<2];
ppp ret;
ppp operator +(const ppp &x,const ppp &y)
{
ret.len=x.len+y.len;
memset(ret.a,0,sizeof(ret.a));
int i,j;
for(i=0;i<=20;i++)
for(j=0;j<=20-i;j++)
ret.a[i+j]=((long long)x.a[i]*y.a[j]+ret.a[i+j])%mod;
return ret;
}
void pup(int x){w[x]=w[x<<1]+w[x<<1|1];}
void pudrev(int x)
{
int i;
rev[x]^=1;
t[x]=(mod-t[x])%mod;
for(i=1;i<=19;i+=2)
w[x].a[i]=(mod-w[x].a[i])%mod;
}
int ksm(int d,int c)
{
int ret=1;
while(c)
{
if(c&1) ret=(long long)ret*d%mod;
d=(long long)d*d%mod;c/=2;
}
return ret;
}
void pudadd(int x,int v)
{
t[x]=(t[x]+v)%mod;
int i,j,l=w[x].len;
for(i=min(20,l);i>=1;i--)
for(j=i-1;j>=0;j--)
w[x].a[i]=((long long)ksm(v,i-j)*C[l-j][i-j]%mod*w[x].a[j]+w[x].a[i])%mod;
}
void pud(int x)
{
int i,j;
if(rev[x])
{
pudrev(x<<1);
pudrev(x<<1|1);
rev[x]=0;
}
if(t[x])
{
pudadd(x<<1,t[x]);
pudadd(x<<1|1,t[x]);
t[x]=0;
}
}
void build(int now,int ll,int rr)
{
l[now]=ll;r[now]=rr;
if(ll==rr)
{
w[now].len=1;
w[now].a[0]=1;
w[now].a[1]=a[ll];
return;
}
int mid=(ll+rr)>>1;
build(now<<1,ll,mid);
build(now<<1|1,mid+1,rr);
pup(now);
}
void change(int now,int ll,int rr,int v)
{
if(l[now]==ll&&r[now]==rr)
{
pudadd(now,v);
return;
}
pud(now);
int mid=(l[now]+r[now])>>1;
if(rr<=mid) change(now<<1,ll,rr,v);
else if(ll>mid) change(now<<1|1,ll,rr,v);
else change(now<<1,ll,mid,v),change(now<<1|1,mid+1,rr,v);
pup(now);
}
void reverse(int now,int ll,int rr)
{
if(l[now]==ll&&r[now]==rr)
{
pudrev(now);
return;
}
pud(now);
int mid=(l[now]+r[now])>>1;
if(rr<=mid) reverse(now<<1,ll,rr);
else if(ll>mid) reverse(now<<1|1,ll,rr);
else reverse(now<<1,ll,mid),reverse(now<<1|1,mid+1,rr);
pup(now);
}
ppp check(int now,int ll,int rr)
{
if(ll==l[now]&&rr==r[now]) return w[now];
pud(now);
int mid=(l[now]+r[now])>>1;
if(rr<=mid) return check(now<<1,ll,rr);
else if(ll>mid) return check(now<<1|1,ll,rr);
else return check(now<<1,ll,mid)+check(now<<1|1,mid+1,rr);
}
int main()
{
int n,q;
scanf("%d%d",&n,&q);
int i,j,x,y,z;
C[0][0]=1;
for(i=1;i<=n;i++)
{
C[i][0]=1;
for(j=1;j<=20;j++)
C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
}
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,1,n);
char s[10];
while(q--)
{
scanf("%s%d%d",s,&x,&y);
if(s[0]=='I')
{
scanf("%d",&z);
change(1,x,y,(z%mod+mod)%mod);
}
else if(s[0]=='R') reverse(1,x,y);
else
{
scanf("%d",&z);
printf("%d\n",check(1,x,y).a[z]);
}
}
}