「TJOI2018」数学计算(线段树)

描述
小豆现在有一个数 x x x ,初始值为 1 1 1 。 小豆有 Q Q Q 次操作,操作有两种类型:
1 1 1 $ m$ : x = x × m x=x×m x=x×m ,输出 x x x m o d mod mod M M M
$2 $ p o s pos pos x = x / x=x/ x=x/ 第$ pos$ 次操作所乘的数(保证第 p o s pos pos 次操作一定为类型 1 1 1,对于每一个类型 1 1 1 的操作至多会被除一次),输出 x x x m o d mod mod M M M

输入
一共有 t t t 组输入。
对于每一组输入,第一行是两个数字 Q , M Q,M Q,M
接下来 Q Q Q 行,每一行为操作类型 o p op op ,操作编号或所乘的数字 m m m (保证所有的输入都是合法的)。

输出
对于每一个操作,输出一行,包含操作执行后的 x x x m o d mod mod M M M 的值。

样例输入
1
10 1000000000
1 2
2 1
1 2
1 10
2 3
2 4
1 6
1 7
1 12
2 7

样例输出
2
1
2
20
10
1
6
42
504
84

对于 20 20% 20的数据,$ 1\leq Q \leq 500 $ ;
对于 100 100% 100 的数据, 1 ≤ Q ≤ 1 0 5 , t ≤ 5 , M ≤ 1 0 9 1\leq Q \leq 10^5 , t\leq 5 , M \leq 10^9 1Q105,t5,M109

这个题可以看出是一道用线段树维护根节点区间乘积的题,由于 Q Q Q的值很小,所以可以设线段树的根节点的左右指针为 T [ 1 ] . l = 1 , T [ 1 ] . r = Q T[1].l=1,T[1].r=Q T[1].l=1,T[1].r=Q,当最初没有修改时,所有叶子结点的值都为 1 1 1,那么整个区间的乘积就为 1 1 1,即题上的 x x x,对于操作 1 1 1,我们让第 i i i个叶节点对应第 i i i次操作,如果第 i i i次操作为操作 1 1 1,那么我们将第 i i i个叶节点的值改为 m m m,否则第 i i i次操作是操作 2 2 2,此时我们只需要将第 p o s pos pos个根节点改回 1 1 1即可。

具体的代码实现:

#include
#define lc (p<<1)
#define rc (p<<1|1)
#define mid (T[p].l+T[p].r>>1)
#define N 100005
using namespace std;
int t,n;
long long mod;
inline long long read(){
	long long ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+ch-'0',ch=getchar();
	return ans;
}
struct Node{int l,r;long long mul;}T[N<<2];
inline void pushup(int p){T[p].mul=T[lc].mul*T[rc].mul%mod;}
inline void build(int p,int l,int r){
	T[p].l=l,T[p].r=r,T[p].mul=1;
	if(l==r)return;
	build(lc,l,mid);
	build(rc,mid+1,r);
}
inline void update(int p,int k,int v){
	if(T[p].l==T[p].r){
		T[p].mul=v;
		return;
	}
	if(k<=mid)update(lc,k,v);
	else update(rc,k,v);
	pushup(p);
}
int main(){
	t=read();
	while(t--){
		n=read(),mod=read();
		build(1,1,n);
		for(int i=1;i<=n;++i){
			int op=read();
			if(op==1){long long v=read();update(1,i,v);}
			else{int pos=read();update(1,pos,1);}
			printf("%lld\n",T[1].mul);
		}
	}
	return 0;
}

你可能感兴趣的:(#,线段树)