Cod sursa(job #2955533)

Utilizator MihiBluBalau Mihai MihiBlu Data 17 decembrie 2022 12:18:56
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.35 kb
#include <fstream>
using namespace std;

ifstream f("datorii.in");
ofstream g("datorii.out");

int n, m, x, poz, q, a, b, sum;
int arb[2600];

void update(int nod, int st, int dr)
{
    if(st==dr)
    {
        arb[nod]=x;
        return;
    }
    int mij=(st+dr)/2;
    if(poz<=mij)
        update(nod*2, st, mij);
    else
        update(nod*2+1, mij+1, dr);
    arb[nod] = arb[nod*2] + arb[nod*2+1];
}

void update2(int nod, int st, int dr)
{
    if(st==dr)
    {
        arb[nod]-=x;
        return;
    }
    int mij=(st+dr)/2;
    if(poz<=mij)
        update(nod*2, st, mij);
    else
        update(nod*2+1, mij+1, dr);
    arb[nod] = arb[nod*2] + arb[nod*2+1];
}

void Querry (int nod, int st, int dr)
{
    if (a <= st && b >= dr)
    {
        sum+=arb[nod];
        return;
    }

    int mij = (st + dr) / 2;

    if (a <= mij) Querry(nod*2, st, mij);

    if (b > mij) Querry(nod*2+1, mij+1, dr);
}



int main()
{
    f>>n>>m;
    for(int i=1; i<=n; i++)
    {
        f>>x;
        poz=i;
        update(1, 1, n);
    }
    while(m--)
    {
        f>>q>>a>>b;
        if(q==0)
        {
            poz=a, x=b;
            update2(1, 1, n);
        }
        if(q==1)
        {
            sum=0;
            Querry(1, 1, n);
            g<<sum<<'\n';
        }
    }
    return 0;
}