Description
有一个长度为n的序列,有三个操作1.I a b c表示将[a,b]这一段区间的元素集体增加c,2.R a b表示将[a,b]区间内所有元素变成相反数,3.Q a b c表示询问[a,b]这一段区间中选择c个数相乘的所有方案的和mod 19940417的值。
100%的数据n<=50000,q<=50000,初始序列的元素的绝对值<=109,I a b c中保证[a,b]是一个合法区间,|c|<=109,R a b保证[a,b]是个合法的区间。Q a b c中保证[a,b]是个合法的区间1<=c<=min(b-a+1,20)。
Solution
计fi为选出i个的答案,那么显然有
f
n
=
∑
i
=
0
n
f
i
×
f
n
−
i
f_n=\sum_{i=0}^nf_i\times f_{n-i}
fn=∑i=0nfi×fn−i
考虑用线段树维护这个东西。合并直接c^2卷积,问题在于取反和区间加怎么搞
观察这个柿子,我们发现对于n为偶数的情况取反不影响,对于n为奇数的情况直接取反就可以了
考虑怎么区间加,记区间长度为len,要加上v。容易得到
f
n
=
∑
i
=
0
n
f
n
−
i
×
v
i
×
(
l
e
n
−
n
+
i
i
)
f_n=\sum_{i=0}^nf_{n-i}\times v^i\times\binom{len-n+i}{i}
fn=∑i=0nfn−i×vi×(ilen−n+i),即我们枚举选i个v,剩余n-i个从剩下len-(n-i)个里面选。实在不行把
∏
(
a
i
+
v
)
\prod(a_i+v)
∏(ai+v)这个东西拆开也能得到这个结论
然后就可以了。比较难写,喜闻乐见(我)要卡常
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)
#define drp(i,st,ed) for (register int i=st;i>=ed;--i)
typedef long long LL;
const int MOD=19940417;
const int N=50005;
bool rev[N<<2];
LL c[N][21],tag[N<<2];
inline void mod(LL &x) {
(x>=MOD)?(x-=MOD):0;
(x<0)?(x+=MOD):0;
}
struct wjp {
LL f[21],size;
wjp() {rep(i,1,20) f[i]=0; f[0]=size=1;}
wjp operator *(wjp b) const {
wjp ret; ret.size=size+b.size;
int tmp=std:: min(20LL,ret.size);
rep(i,1,tmp) rep(j,0,i) {
mod(ret.f[i]+=f[j]*b.f[i-j]%MOD);
}
return ret;
}
wjp operator +(LL v) const {
wjp ret; ret.size=size;
drp(i,std:: min(20LL,size),1) {
LL tmp=1;
rep(j,0,i) {
mod(ret.f[i]+=f[i-j]*tmp%MOD*c[size-i+j][j]%MOD);
tmp=tmp*v%MOD;
}
}
return ret;
}
} rec[N<<2];
inline int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
void push_down(int now) {
if (rev[now]) {
for (int i=1;i<=20;i+=2) {
mod(rec[now<<1].f[i]=-rec[now<<1].f[i]);
mod(rec[now<<1|1].f[i]=-rec[now<<1|1].f[i]);
}
rev[now]=0; rev[now<<1]^=1; rev[now<<1|1]^=1;
mod(tag[now<<1]=-tag[now<<1]); mod(tag[now<<1|1]=-tag[now<<1|1]);
}
if (tag[now]) {
LL w=tag[now]; tag[now]=0;
mod(tag[now<<1]+=w); mod(tag[now<<1|1]+=w);
rec[now<<1]=rec[now<<1]+w; rec[now<<1|1]=rec[now<<1|1]+w;
}
}
void modify(int now,int tl,int tr,int l,int r,LL v) {
if (tl==l&&tr==r) {
if (v==1e9+7) {
for (int i=1;i<=20;i+=2) mod(rec[now].f[i]=-rec[now].f[i]);
rev[now]^=1; mod(tag[now]=-tag[now]);
} else {
rec[now]=rec[now]+v;
mod(tag[now]+=v);
}
return ;
}
push_down(now);
int mid=(tl+tr)>>1;
if (r<=mid) modify(now<<1,tl,mid,l,r,v);
else if (l>mid) modify(now<<1|1,mid+1,tr,l,r,v);
else {
modify(now<<1,tl,mid,l,mid,v);
modify(now<<1|1,mid+1,tr,mid+1,r,v);
}
rec[now]=rec[now<<1]*rec[now<<1|1];
}
wjp query(int now,int tl,int tr,int l,int r) {
if (tl==l&&tr==r) return rec[now];
push_down(now);
int mid=(tl+tr)>>1;
if (r<=mid) return query(now<<1,tl,mid,l,r);
if (l>mid) return query(now<<1|1,mid+1,tr,l,r);
wjp qx=query(now<<1,tl,mid,l,mid);
wjp qy=query(now<<1|1,mid+1,tr,mid+1,r);
return qx*qy;
}
void build(int now,int tl,int tr) {
if (tl==tr) {
LL x=read(); mod(x%=MOD);
rec[now].f[1]=x;
return ;
}
int mid=(tl+tr)>>1;
build(now<<1,tl,mid);
build(now<<1|1,mid+1,tr);
rec[now]=rec[now<<1]*rec[now<<1|1];
}
int main(void) {
freopen("data.in","r",stdin);
freopen("myp.out","w",stdout);
int n=read(),m=read();
c[0][0]=c[1][0]=c[1][1]=1;
rep(i,2,n) {
c[i][0]=1;
rep(j,1,20) {
c[i][j]=c[i-1][j]+c[i-1][j-1];
mod(c[i][j]);
}
}
build(1,1,n);
for (int opt,l,r,x;m--;) {
for (opt=getchar();opt!='Q'&&opt!='R'&&opt!='I';) opt=getchar();
l=read(),r=read(),x;
if (opt=='Q') { x=read();
printf("%lld\n", query(1,1,n,l,r).f[x]);
} else if (opt=='I') { x=read();
modify(1,1,n,l,r,x);
} else modify(1,1,n,l,r,1e9+7);
}
return 0;
}