2020 ICPC·小米邀请赛 决赛热身赛 C.Rikka with Prefix Sum(前缀和/差分/组合数)

题目

t<=3组样例，每组对于长度n(n<=1e5)的序列a[]，m(m<=1e5)次操作，操作分三种

a[]序列初始全部为0

1 l r w 对于[l,r]区间的值+w(0<=w=1e9)

2 对序列a[]做一遍前缀和得到sum[]，然后a[]=sum[]，即对每个i来说，a[i]=sum[i]

3 l r 询问当前ai区间[l,r]的和，答案模998244353，保证3操作数不超过500

题解

晚上回宾馆之后，被自为风月马前卒大佬口胡了一下，发现会了

考虑2搞出的系数其实是一个组合数，

可以根据从左到右前缀和的第x项，当前被做了y次前缀和，

有了这两个值x和y，可以O(1)求出，例如

1 1 1 1
1 2 3 4 //C(2,1) C(3,2) C(4,3)
1 3 6 10 //C(3,1) C(4,2) C(5,3)
1 4 10 20 C//(4,1) C(5,2) C(6,3)

然后考虑[l,r]的+w，对当前求和的影响，

3类型操作不超过500，显然是让我们暴力算每次的贡献

 

假设当前询问[l2,r2]，之前的某次[l1,r1]的+w，从那时起到现在已经做了add次前缀和

则[l1,r1]区间+w，相当于[l1,n]加w，[r1+1,n]减w，

差分之后给l1,r1+1分别打标记，求值的时候多求一次前缀和就求回来了，

对于[l2,r2]内的每个值，对于一个差分标记[x,n]加w来说，

实际应该执行如下代码，

for(int i=l2;i<=r2;++i){
    ans=(ans+C(i-l1+add,i-l1-1))%mod;
}

显然暴力会超时，注意到求区间内组合数*w的区间和，

可以再看做是求了一次前缀和，即把3操作看成是2操作了之后两个单点的值，

答案就是C(r2-l1+add+1,r2-l1-1+1)-C(l2-1+add+1,l2-1+l1+1)了，

另一个标记同理

代码

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353,N=2e5+10;
int t,op,n,m,l,r,w,now,ans;
int fac[N],finv[N];
struct node{
	int w,l,r,num;
};
int C(int n,int m){
	if(n<0 || m<0 || n<m)return 0;
	return 1ll*fac[n]*finv[m]%mod*finv[n-m]%mod;
}
vector<node>q;
int cal(node &x,int l,int r){
	int ans=0,add=now-x.num,w=x.w;//已经被add过几次了 再做一次前缀和
	//+ [max(x.l,l),r] 
	ans=(ans+mod-1ll*C(l-x.l+add,l-x.l-1)*w%mod)%mod;
	ans=(ans+1ll*C(r-x.l+1+add,r-x.l)*w%mod)%mod; 
	//-[max(x.r+1,l),r]
	ans=(ans+mod-1ll*C(l-x.r-1+add,l-x.r-2)*(mod-w)%mod)%mod;
	ans=(ans+1ll*C(r-x.r+add,r-x.r-1)*(mod-w)%mod)%mod;
	//printf("ans:%d\n",ans);
	return ans;
}
int main(){
	finv[0]=finv[1]=1;
	for(int i=2;i<N;++i){
		finv[i]=1ll*(mod-mod/i)*finv[mod%i]%mod;
	}
	fac[0]=fac[1]=1;
	for(int i=2;i<N;++i){
		finv[i]=1ll*finv[i-1]*finv[i]%mod;
		fac[i]=1ll*fac[i-1]*i%mod;
	}
	scanf("%d",&t);
	while(t--){
		q.clear();
		now=0;
		scanf("%d%d",&n,&m);
		while(m--){
			scanf("%d",&op);
			if(op==1){
				scanf("%d%d%d",&l,&r,&w);
				q.push_back(node{w,l,r,now});
			}
			else if(op==2){
				now++;
			}
			else if(op==3){
				scanf("%d%d",&l,&r);
				ans=0;
				for(auto &x:q){
					ans=(ans+cal(x,l,r))%mod;	
				} 
				printf("%d\n",ans);
			}
		}
	} 
	return 0;
}
/*
1 1 1
1 2 3 4 C(2,1)
1 3 6 9 C(3,1) C(4,2)
1 4 10 14 C(4,1) C(5,2)
*/