Cod sursa(job #2999511)

Utilizator Iordache_CezarIordache Cezar Iordache_Cezar Data 11 martie 2023 09:43:23
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <bits/stdc++.h>
#define NMAX 16000

using namespace std;
ifstream fin ("datorii.in");
ofstream fout ("datorii.out");

int ST[4*NMAX];

void update(int nod,int st, int dr, int poz,int val);
int query(int nod, int st, int dr, int qst, int qdr);

int main()
{
    int n, m, poz, val, x, y, tip;
    fin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        fin >> x;
        update(1, 1, n, i, -x);
    }
    for (int i = 1; i <= m; i++)
    {
        fin >> tip;
        if (tip == 1)
        {
            fin >> x >> y;
            fout << query(1,1, n, x, y) << '\n';
        }
        else
        {
            fin >> poz >> val;
            update(1,1, n, poz, val);
        }
    }
    return 0;
}

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

int query(int nod, int st, int dr, int qst, int qdr)
{
    if (qst <= st && dr <= qdr)
        return ST[nod];
    int mij = (st + dr) / 2, s1 = 0, s2 = 0;
    if (qst <= mij)
        s1 = query(2*nod, st, mij, qst, qdr);
    if (mij < qdr)
        s2 = query(2*nod+1, mij+1, dr, qst, qdr);
    return s1 + s2;
}