Cod sursa(job #2452413)

Utilizator stefantagaTaga Stefan stefantaga Data 30 august 2019 17:10:09
Problema Hotel Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.05 kb
#include <fstream>

using namespace std;
ifstream f("hotel.in");
ofstream g("hotel.out");
struct wow
{
    int a,b,sol;
}lazy[400005],arb[400005];
void initializare (int nod, int st, int dr)
{
    arb[nod].a=arb[nod].b=arb[nod].sol=dr-st+1;

    if(st==dr)
        return;

    int mij=(st+dr)/2;

    initializare (2*nod, st, mij);
    initializare (2*nod+1, mij+1, dr);
}
void propaga (int nod,int st,int dr)
{
    if (st==dr)
    {
        return ;
    }
    if (lazy[nod].a==-1)
    {
        return ;
    }
    arb[2*nod]=arb[2*nod+1]=lazy[nod];
    lazy[2*nod]=lazy[2*nod+1]=lazy[nod];
    lazy[nod]={-1,-1,-1};
}
void update (int st,int dr,int nod ,int ua ,int ub,int val)
{
    if (ua<=st&&dr<=ub)
    {
        if (val==0)
        {
            arb[nod].a=dr-st+1;
            arb[nod].b=dr-st+1;
            arb[nod].sol=dr-st+1;
        }
        else
        {
            arb[nod]={0,0,0};
        }
        lazy[nod]=arb[nod];
        return ;
    }
    propaga(nod,st,dr);
    int mij=(st+dr)/2;
    if (ua<=mij)
    {
        update(st,mij,2*nod,ua,ub,val);
    }
    if (ub>mij)
    {
        update(mij+1,dr,2*nod+1,ua,ub,val);
    }
    arb[nod].sol=max(arb[2*nod].b+arb[2*nod+1].a,max(arb[2*nod].sol,arb[2*nod+1].sol));
    arb[nod].a = arb[2 * nod].a;
    if (arb[nod].a == mij - st + 1) arb[nod].a += arb[2 * nod + 1].a;
    arb[nod].b = arb[2 * nod + 1].b;
    if (arb[nod].b == dr - mij) arb[nod].b += arb[2 * nod].b;
}
int n,q,i,tip,st,nr,dr;
int main()
{
    f>>n>>q;
    for (i=1;i<=4*n;i++)
    {
        lazy[i]={-1,-1,-1};
    }
    initializare(1,1,n);
    for (i=1;i<=q;i++)
    {
        f>>tip;
        if (tip==3)
        {
            propaga(1,1,n);
            g<<arb[1].sol<<'\n';
        }
        else
        if (tip==2)
        {
            f>>st>>nr;
            dr=st+nr-1;
            update(1,n,1,st,dr,0);
        }
        else
        {
            f>>st>>nr;
            dr=st+nr-1;
            update(1,n,1,st,dr,1);
        }
    }
    return 0;
}