P4588 [TJOI2018] 數學計算

原題連結

題解

由於模擬會爆 longlong，所以用線段樹維護每次操作的值，初始每次操作的值均為1

操作一令對應節點變為 m

操作二令對應節點變為 1

返回整棵樹的值(相乘)

code

#include<bits/stdc++.h>
#define ll long long
using namespace std;

ll tree[400005];
ll q,mod;

void update(int node,int l,int r,int id,int v)
{
    if(l>id||r<id) return;

    if(l==id&&r==id)
    {
        tree[node]=v%mod;
        return;
    }

    int mid=(l+r)/2;
    update(2*node,l,mid,id,v);
    update(2*node+1,mid+1,r,id,v);

    tree[node]=tree[node*2]*tree[node*2+1]%mod;
}


void solve()
{
    cin>>q>>mod;

    fill(tree,tree+4*q+1,1);

    ll x=1;
    for(int i=1;i<=q;i++)
    {
        int op;
        cin>>op;
        if(op==1)
        {
            int m;
            cin>>m;

            update(1,1,q,i,m);
            cout<<tree[1]<<'\n';
        }
        else
        {
            int pos;
            cin>>pos;
            update(1,1,q,pos,1);
            cout<<tree[1]<<'\n';
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t=1;
    cin>>t;
    while(t--) solve();
    return 0;
}