Cod sursa(job #1083309)

Utilizator ionutz_cnnbIonutz cnnb ionutz_cnnb Data 15 ianuarie 2014 21:01:58
Problema Hotel Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.83 kb
// Eu - Hotel O(n+m*log n)
#include <fstream>
using namespace std;
#define Max (1<<18)
#define mij (st + dr) / 2
struct nod_arb {int st0; int dr0; int tot0; int upd;} A[Max];
int max(int a,int b)
{
    return a>b ? a:b;
}
void build(int nod,int st,int dr)
{
    A[nod].st0=A[nod].dr0=A[nod].tot0=dr-st+1;
    A[nod].upd=-1;
    if(st < dr)
    {
        build(2*nod, st, mij);
        build(2*nod+1, mij+1, dr);
    }
}
 void update(int nod,int st,int dr,int a,int b,int v)
{
    // daca intervalul [st,dr] e inclus in [a,b] se updateaza tot
    if(a <= st && b >= dr)
    {
        A[nod].upd = v;
        A[nod].st0=A[nod].dr0=A[nod].tot0=v*(dr-st+1);
        return;
    }
    // daca intervalul curent e deja updatat iar cel inferior nu, se updateaza un nivel
    if(A[nod].upd != -1)
    {
        update(2*nod, st, mij, st , mij, A[nod].upd);
        update(2*nod+1, mij+1, dr, mij+1, dr, A[nod].upd);
        A[nod].upd = -1;
    }
    if(a <= mij)
        update(2*nod, st, mij, a, b, v);
    if(b > mij)
        update(2*nod+1, mij+1, dr, a, b, v);
    // se updateaza fiecare interval din nod
    A[nod].tot0 = max(max(A[2*nod].tot0, A[2*nod+1].tot0) , A[2*nod].dr0 + A[2*nod+1].st0);
    if(A[2*nod+1].dr0 == dr - mij)
        A[nod].dr0 = A[2*nod].dr0 + A[2*nod+1].dr0;
    else
        A[nod].dr0 = A[2*nod+1].dr0;
    if(A[2*nod].st0 == mij - st + 1)
        A[nod].st0 = A[2*nod].st0 + A[2*nod+1].st0;
    else
        A[nod].st0 = A[2*nod].st0;
}
 int main()
{
    int i,n,m,op,x,y;
    ifstream fi("hotel.in");
    ofstream fo("hotel.out");
    fi>>n>>m;
    build(1,1,n);
    for(i=1;i<=m;i++)
    {
        fi>>op;
        if(op == 3)
            fo<<A[1].tot0<<"\n";
        else
        {
            fi>>x>>y;
            update(1,1,n,x,x+y-1,op-1);
        }
    }
    return 0;
}