[SHOI2016]随机序列 线段树

[SHOI2016]

  • 写几组例子可以发现对答案产生贡献的只有一段前缀的连乘,因为只要从这个地方断开就会有+,-两种符号相互抵消,对答案的贡献就是0,然后问题就变成了区间修改,线段树即可
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
using namespace std;
const int mod=1e9+7;
const int N=1e6;
inline int read(){
	int num=0;char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))num=num*10+ch-'0',ch=getchar();
	return num;
}
inline void Print(int x){if(x>9)Print(x/10);putchar(x%10+'0');}
int power(int a,int b)
{int ans=1;for(;b;b>>=1,a=1LL*a*a%mod)if(b&1)ans=1LL*ans*a%mod;return ans;}
int P[N],s[N],a[N],sum[N],mul[N],ans,n,m,p,v;
void build(int o,int L,int R){
	int lc=o<<1,rc=o<<1|1,M=L+R>>1;
	mul[o]=1;
	if(L==R){
		sum[o]=s[L];
		return;
	}
	build(lc,L,M),build(rc,M+1,R);
	sum[o]=(sum[lc]+sum[rc])%mod;
}
void pushdown(int o,int L,int R){
	int lc=o<<1,rc=o<<1|1;
	if(mul[o]!=1){
		mul[lc]=1LL*mul[lc]*mul[o]%mod;
		mul[rc]=1LL*mul[rc]*mul[o]%mod;
		sum[lc]=1LL*sum[lc]*mul[o]%mod;
		sum[rc]=1LL*sum[rc]*mul[o]%mod;
		mul[o]=1;
	}
}
void update(int o,int L,int R,int ql,int qr,int v){
	int lc=o<<1,rc=o<<1|1,M=L+R>>1;
	if(ql<=L&R<=qr){
		mul[o]=1LL*mul[o]*v%mod;
		sum[o]=1LL*sum[o]*v%mod;
		return ;
	}
	pushdown(o,L,R);
	if(ql<=M)update(lc,L,M,ql,qr,v);
	if(M<qr)update(rc,M+1,R,ql,qr,v);
	sum[o]=(sum[lc]+sum[rc])%mod;
}
int main()
{
	//freopen("a.in","r",stdin);
	//freopen("a.out","w",stdout);
	n=read(),m=read();
	rep(i,1,n)a[i]=read();
	P[0]=1;rep(i,1,n)P[i]=1LL*3*P[i-1]%mod;
	s[0]=1;rep(i,1,n)s[i]=1LL*a[i]*s[i-1]%mod;
	rep(i,1,n-1)s[i]=1LL*s[i]*2%mod*P[n-i-1]%mod;
	build(1,1,n);
	while(m--){
		p=read(),v=read();
		if(a[p]==0){
			ans=0;
			a[p]=v;
			s[0]=1;rep(i,1,n)s[i]=1LL*a[i]*s[i-1]%mod;
			rep(i,1,n-1)s[i]=1LL*s[i]*2%mod*P[n-i-1]%mod;
			build(1,1,n);
			rep(i,1,n)ans=(ans+s[i])%mod;
			Print(ans);puts("");
		}else{
			int inv=power(a[p],mod-2); a[p]=v;
			update(1,1,n,p,n,1LL*v*inv%mod);
			Print(sum[1]);puts("");
		}
	}return 0;
}

猜你喜欢

转载自blog.csdn.net/strangeDDDF/article/details/88787139
今日推荐