Cod sursa(job #2480315)

Utilizator Vasilescu_CosminVasilescu Cosmin Vasilescu_Cosmin Data 25 octombrie 2019 12:05:18
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.65 kb
#include <bits/stdc++.h>

using namespace std;

ifstream in ("arbint.in");
ofstream out ("arbint.out");

int N, M;
int MaxArb[400066];
int start, finish, Val, Pos, suma;

void Update1(int nod, int left, int right)
{
     if ( left == right )
     {
          MaxArb[nod] = Val;
          return;
     }
     int div = (left+right)/2;
     if ( Pos <= div )
        Update1(2*nod,left,div);
     else
        Update1(2*nod+1,div+1,right);
     MaxArb[nod] = MaxArb[2*nod]+MaxArb[2*nod+1];
}

void Update2(int nod, int left, int right)
{
     if ( left == right )
     {
          MaxArb[nod] -= Val;
          return;
     }
     int div = (left+right)/2;
     if ( Pos <= div )
        Update2(2*nod,left,div);
     else
        Update2(2*nod+1,div+1,right);
     MaxArb[nod] = MaxArb[2*nod]+MaxArb[2*nod+1];
}

void Query(int nod, int left, int right)
{
     if ( start <= left && right <= finish )
     {
          suma += MaxArb[nod];
          return;
     }
     int div = (left+right)/2;
     if ( start <= div )
        Query(2*nod,left,div);
     if ( div < finish )
        Query(2*nod+1,div+1,right);
}

int main()
{
    int X, A, B;
    in>>N>>M;
    for ( int i = 1; i <= N; i++ )
    {
        in>>X;
        Pos = i, Val = X;
        Update1(1,1,N);
    }

    for ( int i = 1; i <= M; i++ )
    {
        in>>X>>A>>B;
        if ( X == 1 )
        {
             suma  = 0;
             start = A, finish = B;
             Query(1,1,N);
             out<<suma<<'\n';
        }
        else
        {
            Pos = A, Val = B;
            Update2(1,1,N);
        }
    }
    return 0;
}