Cod sursa(job #2903895)

Utilizator bianca.andreiAndrei Bianca bianca.andrei Data 17 mai 2022 21:18:27
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.39 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");


int arb[4000000], v[1000000];

void creare(int x, int st, int dr)
{
    if (st == dr)
    {
        arb[x] = v[st];
        return;
    }
    creare(x*2, st, (st+dr)/2);
    creare(x*2+1, (st+dr)/2+1, dr);
    arb[x] = arb[x*2] + arb[x*2+1];
}

void modificare(int x, int st, int dr, int i, int val)
{
    if(i<st||i>dr)
        return;
    if(st == dr)
    {
        arb[x] -= val;
        return;
    }
    int mijloc = (st+dr)/2;
    if(i<=mijloc)
    modificare(x*2, st, mijloc, i, val);
    else
    modificare(x*2+1, mijloc+1, dr, i, val);
    arb[x] = arb[x*2] + arb[x*2+1];
}

int suma(int x, int st, int dr, int i, int j)
{
    if(i<=st && j>=dr)
        return arb[x];
    if(dr<i || st>j)
    {
        //arb[x] -= val;
        return 0;
    }
    int mijloc = (st+dr)/2;
    return suma(x*2, st, mijloc, i, j) + suma(x*2+1, mijloc+1, dr, i, j);
}

int main()
{
    int n, i, operatie, a, b, m;

    f>>n>>m;

    for(i=1;i<=n;i++)
    {
        f>>v[i];
    }

    creare(1, 1, n);


    for(i=0;i<m;i++)
    {
        f>>operatie>>a>>b;

        if(operatie==0)
        {
            modificare(1, 1, n, a, b);
        }
        else
        {
            g<<suma(1, 1, n, a, b)<<endl;
        }
    }

    return 0;
}