Cod sursa(job #2613713)

Utilizator robert.barbu27robert barbu robert.barbu27 Data 10 mai 2020 15:40:08
Problema Hotel Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.56 kb
#include <bits/stdc++.h>
#define INF 1000000000
#define ll long long int
using namespace std;
ifstream f("hotel.in");
ofstream g("hotel.out");
struct SegmentTree
{
    int pref,suf,vmax,lazy;
};
SegmentTree aint[800009];
void build(int nod,int st,int dr)
{
    if(st==dr)
    {
        aint[nod].pref=aint[nod].suf=aint[nod].vmax=1;
        return ;
    }
    int mij=(st+dr)/2;
    build(2*nod,st,mij);
    build(2*nod+1,mij+1,dr);
    aint[nod].pref=aint[nod].suf=aint[nod].vmax=aint[2*nod].vmax+aint[2*nod+1].vmax;
}
void update(int nod,int st,int dr,int tip,int ql,int qr)
{
    //cout<<nod<<" ";
    if(st>dr) return;
    if(aint[nod].lazy==1)
    {
        aint[nod].pref=0;
        aint[nod].suf=0;
        aint[nod].vmax=0;
        if(st!=dr)
        {
            aint[2*nod].lazy=aint[2*nod+1].lazy=1;

        }
        aint[nod].lazy=-1;
    }
    else
    {
        if(aint[nod].lazy==0)
        {
            aint[nod].suf=dr-st+1;
            aint[nod].pref=dr-st+1;
            aint[nod].vmax=dr-st+1;
            if(st!=dr)
            {
                 aint[2*nod].lazy=aint[2*nod+1].lazy=0;
            }
            aint[nod].lazy=-1;
        }
    }
    if(st>qr||dr<ql) return;
    if(ql<=st&&dr<=qr)
    {
        if(tip==1)
        {
            aint[nod].pref=aint[nod].suf=aint[nod].vmax=0;
            if(st!=dr)
            {
                aint[2*nod].lazy=aint[2*nod+1].lazy=1;
            }
        }
        else
        {
            aint[nod].suf=dr-st+1;
            aint[nod].pref=dr-st+1;
            aint[nod].vmax=dr-st+1;
            if(st!=dr)
            {
                aint[2*nod].lazy=aint[2*nod+1].lazy=0;
            }
        }
        return;
    }
    int mij=(st+dr)/2;
    update(2*nod,st,mij,tip,ql,qr);
    update(2*nod+1,mij+1,dr,tip,ql,qr);
    aint[nod].vmax=max(aint[2*nod].vmax,aint[2*nod+1].vmax);
    aint[nod].vmax=max(aint[nod].vmax,aint[2*nod].suf+aint[2*nod+1].pref);
    aint[nod].pref=aint[2*nod].pref;
    aint[nod].suf=aint[2*nod+1].suf;
    if(aint[2*nod].vmax==mij-st+1) aint[nod].pref+=aint[2*nod+1].pref;
    if(aint[2*nod+1].vmax==dr-mij) aint[nod].suf+=aint[2*nod].suf;


}
int query()
{
    return aint[1].vmax;
}

int n,q;
int main()
{
    f>>n>>q;
    build(1,1,n);

    for(int i=1;i<=q;i++)
    {

        int tip;
        f>>tip;
        if(tip==3) g<<query()<<'\n';
        else
        {
            int st,dr;
            f>>st>>dr;
            dr=st+dr-1;

             update(1,1,n,tip,st,dr);

        }
    }

}