【线段树】BZOJ 5334 数学计算

题目内容

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

输入格式

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

输出格式

对于每一个操作，输出一行，包含操作执行后的\(x\ mod\ M\)的值

数据范围

对于\(20\%\)的数据，\(1≤Q≤500\)；
对于\(100\%\)的数据，\(1≤Q≤105\),\(t≤5\),\(M≤109\)。

样例

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

思路

线段树+单点修改。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int Mod;
ll ans[maxn<<2];

void build(int x,int l,int r){
    ans[x]=1ll;
    if(l==r) return;
    int mid=l+r>>1;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
}

void modify(int x,int l,int r,int p,int q){
    if(l==r){
        ans[x]=(ll)q%Mod;
        return;
    }

    int mid=l+r>>1;
    if(p<=mid)modify(x<<1,l,mid,p,q);
    else modify(x<<1|1,mid+1,r,p,q);
    ans[x]=ans[x<<1]*ans[x<<1|1]%Mod;
}

int main(){
    int T,q,p,x;
    scanf("%d",&T);

    while(T--){
        scanf("%d%d",&q,&Mod);
        build(1,1,q);

        for(int i=1;i<=q;i++){
            scanf("%d%d",&p,&x);
            if(p==1)modify(1,1,q,i,x);
            else modify(1,1,q,x,1);
            printf("%lld\n",ans[1]);
        }
    }
    
    return 0;
}