BZOJ5334 [TJOI2018] 数学计算 【线段树分治】

题目分析：

　　大概是考场上的签到题。首先mod不是质数，所以不能求逆元。注意到有加入操作和删除操作。一个很典型的想法就是线段树分治。建立时间线段树然后只更改有影响的节点，最后把所有标记下传。时间复杂度是O(nlogn)。

代码：

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

int q,mod;

int data[405000];
struct node{int l,r,d;}p[102000];

void read(){
    memset(p,0,sizeof(p));
    memset(data,0,sizeof(data));
    scanf("%d%d",&q,&mod);
    for(int i=1;i<=q;i++){
    int cas; scanf("%d",&cas);
    int x; scanf("%d",&x);
    if(cas == 1){
        p[i].l = i;p[i].d = x;
    }else{p[x].r = i;}
    }
}

void add(int now,int tl,int tr,int l,int r,int d){
    if(tl >= l && tr <= r){
    data[now] = (1ll*data[now]*d)%mod;
    return;
    }
    if(tl > r || tr < l) return;
    int mid = (tl+tr)/2;
    add(now<<1,tl,mid,l,r,d);
    add(now<<1|1,mid+1,tr,l,r,d);
}

void dfs(int now,int tl,int tr){
    if(tl == tr){printf("%d\n",data[now]);return;}
    int L = now*2,R = now*2+1;
    data[L] = (1ll*data[L]*data[now])%mod;
    data[R] = (1ll*data[R]*data[now])%mod;
    data[now] = 1;int mid =(tl+tr)/2;
    dfs(L,tl,mid); dfs(R,mid+1,tr);
}

void work(){
    for(int i=1;i<=4*q;i++) data[i] = 1;
    for(int i=1;i<=q;i++){
    if(p[i].l == 0) continue;
    if(p[i].r == 0) add(1,1,q,p[i].l,q,p[i].d);
    else add(1,1,q,p[i].l,p[i].r-1,p[i].d);
    }
    dfs(1,1,q);
}

int main(){
    int t; scanf("%d",&t);
    while(t--){
    read();
    work();
    }
    return 0;
}