牛客寒假算法基础集训营2-F.小沙的算数

标签：乘法逆元

思路

由于运算的优先级，可以发现加号将整个式子分成了多个区间且各区间互不影响，乘号所连接的数则可以划分到一个区间，而且本题中查询的值一直是整个区间，所以可以用一个数组book[i]来标记每个数对应的组，用sum[i]来保存每个组的计算结果，然后将每个组的值加起来就得到了开始整个式子的值ans，当将下标为x的数a[x]改为y时，其对应的组的值sum会变成sum/a[x]*y，而其他组不受影响
由于本题需要取模且有除法运算，所以需要用到乘法逆元，最后还要注意ans取模后可能会比sum小，这样相减就会出现负数，所以在减的时候要先加上模数再取模

代码实现

#include<iostream>
#include<stdio.h>
using namespace std;
const int mod=1e9+7;
const int N=1e6+5;
char s[N];
long long n,q,a[N],book[N],sum[N];
long long qpow(long long x,long long y){
	long long c=1;
	while(y>0){
		if(y&1) c=c*x%mod;
		x=x*x%mod;
		y>>=1;
	}
	return c;
}
int main(){
	cin>>n>>q;
	int t=1;
	for(int i=1;i<n;i++) cin>>s[i];
	for(int i=1;i<=n;i++) cin>>a[i];
	sum[1]=a[1];
	book[1]=1;
	for(int i=2;i<=n;i++){
		if(s[i-1]=='*'){              //遇到乘号归为一组 
			book[i]=t;
			sum[t]=sum[t]*a[i]%mod;   //计算每组的值 
		}else{                        //遇到加号换组 
			sum[++t]=a[i];
			book[i]=t;
		}
	}
	long long ans=0;
	for(int i=1;i<=t;i++) ans=(ans+sum[i])%mod;
	while(q--){
		int x,y;
		cin>>x>>y;
		ans=(ans-sum[book[x]]+mod)%mod;    //加上mod再%mod，防止出现负数 
		sum[book[x]]=sum[book[x]]*qpow(a[x],mod-2)%mod*y%mod;   //逆元计算除法 
		ans=(ans+sum[book[x]])%mod;
		a[x]=y;
		cout<<ans<<endl;
	}
	return 0;
}