P4588 [TJOI2018]数学计算

题目描述

小豆现在有一个数x，初始值为1.小豆有Q次操作，操作有两种类型:
1 m：x=x×m输出x%mod;
2 pos：x= x / 第pos次操作所乘的数（保证第pos次操作一定为类型1,对于每一个类型1的操作至多会被除一次）输出x%mod;

输入格式

一共有t组输入(t≤5);
对于每一组输入,第一 行是两个数字Q,mod(Q≤100000,mod≤100000000);
接下来Q行，每一行为操作类型op,操作编号或所乘的数字m（保证所有的输入都是合法的）.

输出格式

对于每一个操作，输出一行，包含操作执行后的x%mod的值

样例

输入
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%的数据，1≤Q≤500
对于100%100%的数据，1≤Q≤100000

题目分析

我们可以维护一棵[1,Q]的线段树，每一个叶结点代表一次操作，[l,r]的一段区间表示[l,r]区间内的全部操作数的积。因此tr[1]维护的就是当前操作完成后的x的值。
有两个操作：
1：给你一个m，让x=x*m，我们只需要将线段树中i（第i次操作）点的值改为m即可。
2：给你一个数pos，让x=x/第pos次操作所乘的数，因为保证了pos次操作肯定为操作1，因此我们只需要将线段树pos点的权值改为1即可。
每次操作完输出tr[1]的权值即可。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <stack>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#include <iomanip>
#define LL long long
#define PII pair<int,int>
using namespace std;
const int N=1e5+5;
struct Node{
	int l,r;
	int val;	//记录权值
}tr[N*4];
int n,mod;
void pushup(int u)	//两个子段的权值相乘即可得到u段的权值
{
	tr[u].val=(LL)tr[u<<1].val*tr[u<<1|1].val%mod;
}
void build(int u,int l,int r)	//建树（模板）
{
	if(l==r) tr[u]={l,r,1};		//因为维护的权值的操作是相乘，因此默认值为1
	else
	{
		tr[u]={l,r,1};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		pushup(u); 
	}
}
void update(int u,int x,int v)	//修改操作，将x点的值改为v（模板）
{
	if(tr[u].l==x&&tr[u].r==x) tr[u].val=v;
	else
	{
		int mid=tr[u].l+tr[u].r>>1;
		if(mid>=x) update(u<<1,x,v);
		else update(u<<1|1,x,v);
		pushup(u);
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d %d",&n,&mod);
		build(1,1,n); 
		for(int i=1;i<=n;i++)
		{
			int op,x;
			scanf("%d %d",&op,&x);
			if(op==1)
			{
				update(1,i,x);
				printf("%d\n",tr[1].val); 
			}
			else
			{
				update(1,x,1);
				printf("%d\n",tr[1].val); 
			}
		}
	}
	return 0;
}