AcWing 1275. 最大数(线段树）

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int m,p,n;
struct node{
    int l,r,v;
}tr[N<<2];
void build(int u,int l,int r){
    tr[u]={l,r};
    if(l==r)return ;
    int mid=l+r>>1;
    build(u<<1,l,mid),build(u<<1|1,mid+1,r);
}
void pushup(int u){
    tr[u].v=max(tr[u<<1].v,tr[u<<1|1].v);
}
int query(int u,int l,int r){
    if(tr[u].l>=l&&tr[u].r<=r){
        return tr[u].v;
    }
    int mid=tr[u].l+tr[u].r>>1,v=0;
    if(l<=mid)v=query(u<<1,l,r);
    if(r>mid)v=max(v,query(u<<1|1,l,r));
    return v;
}
void modify(int u,int x,int v){
    if(tr[u].l==x&&tr[u].r==x){
        tr[u].v=v;
        return;
    }
    int mid=tr[u].l+tr[u].r>>1;
    if(x<=mid)modify(u<<1,x,v);
    if(x>mid)modify(u<<1|1,x,v);
    pushup(u);
}
int main(){
    scanf("%d%d",&m,&p);
    build(1,1,m);
    char op[2];
    int x,last=0;
    while(m--){
        scanf("%s%d",op,&x);
        if(*op=='A'){
            modify(1,n+1,(last+x)%p);
            n++;
        }
        else{
            last=query(1,n-x+1,n);
            printf("%d\n",last);
        }
    }
}