[清华集训]序列操作

题目

\(\operatorname{NOIP}\)停办的日子发篇博客纪念一下

发现这个\(c\leq \min(20,r-l+1)\),显然可以把\(0\)\(20\)的答案都存一下,线段树合并两个区间就是直接做一个卷积

考虑区间加操作,设\(f_i\)表示当前区间里所有长度为\(i\)的子序列乘积的和,我们考虑每一个子序列对这个的贡献,有

\[f_i=\sum_{j=0}^if_j\binom{len-j}{i-j}c^{i-j} \]

就是考虑一个原来长度为\(j\)的子序列,只有乘上\(c^{i-j}\)才能对长度为\(i\)的子序列产生贡献,同时对于剩下的\(len-j\)个数我们再随便选出\(i-j\)个就能构成长度为\(j\)的子序列了,于是再乘上\(\binom{len-j}{i-j}\)

这个东西也可以直接\(20^2\)算出来

对于取反,我们只需要把长度为奇数的子序列取一个反就好了,同时把加法标记取反

于是就以\(O(20^2m\operatorname{logn})\)的复杂度做完了这道题

代码

#include<bits/stdc++.h>
#define re register
#define LL long long
#define min std::min
inline int read() {
	char c=getchar();int x=0,r=1;
	while(c<'0'||c>'9') {if(c=='-') r=-1;c=getchar();}
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return r*x;
}
const int maxn=5e4+5;
const int mod=19940417;
int c[maxn][21];
int l[maxn*3],r[maxn*3],tag[maxn*3],rev[maxn*3];
int dp[maxn*3][21],h[21],pw[21],ans[21],n,m;
inline int C(int n,int m) {return m>n?0:c[n][m];}
inline void merge(int *f,int *g,int T) {
	for(re int i=0;i<=T;i++) h[i]=0;
	for(re int i=0;i<=T;i++)
		for(re int j=0;i+j<=T;j++)
			h[i+j]=(h[i+j]+1ll*f[i]*g[j]%mod)%mod;
	for(re int i=0;i<=T;i++) f[i]=h[i];
}
inline void pushup(int x) {
	for(re int i=0;i<=20;i++) 
		dp[x][i]=dp[x<<1][i];
	merge(dp[x],dp[x<<1|1],20);
}
void build(int x,int y,int i) {
	l[i]=x,r[i]=y;
	if(x==y) {
		dp[i][0]=1;dp[i][1]=(read()+mod)%mod;return;
	}
	int mid=x+y>>1;
	build(x,mid,i<<1),build(mid+1,y,i<<1|1);
	pushup(i);
}
void solve(int *f,int v,int len) {
	int T=min(20,len);
	for(re int i=1;i<=T;i++) pw[i]=1ll*pw[i-1]*v%mod;
	for(re int i=T;i;--i) 
		for(re int j=i-1;j>=0;--j)
			f[i]=(f[i]+1ll*f[j]*pw[i-j]%mod*C(len-j,i-j)%mod)%mod;
}
inline void pushdown(int x) {
	if(rev[x]) {
		rev[x<<1|1]^=1,rev[x<<1]^=1;
		tag[x<<1|1]=(mod-tag[x<<1|1])%mod;
		tag[x<<1]=(mod-tag[x<<1])%mod;
		for(re int i=1;i<=20;i+=2)
			dp[x<<1][i]=(mod-dp[x<<1][i])%mod,
			dp[x<<1|1][i]=(mod-dp[x<<1|1][i])%mod;
		rev[x]=0;
	}
	if(tag[x]) {
		tag[x<<1]=(tag[x<<1]+tag[x])%mod;
		tag[x<<1|1]=(tag[x<<1|1]+tag[x])%mod;
		solve(dp[x<<1],tag[x],r[x<<1]-l[x<<1]+1);
		solve(dp[x<<1|1],tag[x],r[x<<1|1]-l[x<<1|1]+1);
		tag[x]=0;
	}
}
void add(int x,int y,int v,int i) {
	if(x<=l[i]&&y>=r[i]) {
		solve(dp[i],v,r[i]-l[i]+1);
		tag[i]=(tag[i]+v)%mod;
		return;
	}
	pushdown(i);
	int mid=l[i]+r[i]>>1;
	if(x<=mid) add(x,y,v,i<<1);
	if(y>mid) add(x,y,v,i<<1|1);
	pushup(i);
}
void change(int x,int y,int i) {
	if(x<=l[i]&&y>=r[i]) {
		rev[i]^=1;
		tag[i]=(mod-tag[i])%mod;
		for(re int j=1;j<=20;j+=2)
			dp[i][j]=(mod-dp[i][j])%mod;
		return;
	}
	pushdown(i);
	int mid=l[i]+r[i]>>1;
	if(x<=mid) change(x,y,i<<1);
	if(y>mid) change(x,y,i<<1|1);
	pushup(i);
} 
void query(int x,int y,int v,int i) {
	if(x<=l[i]&&y>=r[i]) {
		merge(ans,dp[i],v);
		return;
	}
	pushdown(i);
	int mid=l[i]+r[i]>>1;
	if(x<=mid) query(x,y,v,i<<1);
	if(y>mid) query(x,y,v,i<<1|1);
}
int main() {
	n=read(),m=read();build(1,n,1);
	for(re int i=0;i<=n;i++) c[i][0]=1;
	for(re int i=1;i<=n;i++)
		for(re int j=1;j<=min(i,20);j++)
			c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
	char op[4];int x,y,v;ans[0]=pw[0]=1;
	while(m--) {
		scanf("%s",op);x=read(),y=read();
		if(op[0]=='I') {
			v=read();v=(v%mod+mod)%mod;
			add(x,y,v,1);
		}
		if(op[0]=='Q') {
			v=read();v=(v%mod+mod)%mod;
			for(re int i=1;i<=20;i++) ans[i]=0;
			query(x,y,v,1);
			printf("%d\n",ans[v]);
		}
		if(op[0]=='R') change(x,y,1);
	}
	return 0;
}
posted @ 2019-08-16 20:24  asuldb  阅读(218)  评论(0编辑  收藏  举报