bzoj2962 序列操作 线段树+组合数学

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={\sum }_{i=0}^n{f}_i\times f_{n-i}$
考虑用线段树维护这个东西。合并直接c^2卷积，问题在于取反和区间加怎么搞
观察这个柿子，我们发现对于n为偶数的情况取反不影响，对于n为奇数的情况直接取反就可以了
考虑怎么区间加，记区间长度为len，要加上v。容易得到$f_n={\sum }_{i=0}^n{f}_{n-i}\times v^i\times \left(\genfrac{}{}{0ex}{}{len-n+i}{i}\right)$，即我们枚举选i个v，剩余n-i个从剩下len-(n-i)个里面选。实在不行把$\prod (a_i+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;
}