Gym - 101889D：Daunting device （老司机树）

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
struct in{
    int L,R,v;
    in(){}
    in(int LL,int RR,int vv):L(LL),R(RR),v(vv){}
    friend bool operator<(in a,in b){
        if(a.L==b.L) return a.R<b.R;
        return a.L<b.L;
    }
};
set<in>s;
#define IT set<in>::iterator
int num[100010],ans;
IT split(int pos)
{
    IT it=s.lower_bound(in(pos,-2,-2));
    if(it!=s.end()&&(*it).L==pos) return it;
    it--;
    int l=(*it).L,r=(*it).R,v=(*it).v;
    s.erase(it);
    s.insert(in(l,pos-1,v));
    return s.insert(in(pos,r,v)).first;
}
void Assign(int L,int R,int X)
{
    IT it2=split(R+1),it1=split(L);
    for(IT it=it1;it!=it2;it++) {
        num[(*it).v]-=((*it).R-(*it).L+1);
    }
    s.erase(it1,it2);
    s.insert(in(L,R,X));
    num[X]+=R-L+1;
}
int main()
{
    int N,L,C,P,X,A,B,S;
    scanf("%d%d%d",&L,&C,&N);
    s.insert(in(0,L-1,1)); num[1]=L;
    rep(i,1,N){
        scanf("%d%d%d%d",&P,&X,&A,&B); A%=L; B%=L;
        S=num[P]%L;
        int l=(A+1LL*S*S%L)%L,r=(A+1LL*(S+B)*(S+B)%L)%L;
        Assign(min(l,r),max(l,r),X);
    }
    rep(i,1,C) ans=max(ans,num[i]);
    printf("%d\n",ans);
    return 0;
}