Cod sursa(job #2903798)

Utilizator bianca2002Bianca bianca2002 Data 17 mai 2022 20:33:54
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.53 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");


int arb[4000001];

void creare(int x, int st, int dr, int k, int val)
{

    if (st == dr)
    {
        arb[x] = val;
        return;
    }

    int mij = (st+dr)/2;

    if(k <= mij) creare(x*2, st, mij, k, val);

    else creare(x*2+1, mij+1, dr, k, val);

    arb[x] = arb[x*2] + arb[x*2+1];
}

void modificare(int x, int st, int dr, int i, int val)
{
    if(st == dr)
    {
        arb[x] -= val;
        return;
    }
    int mij = (st+dr)/2;

    if(i<=mij) modificare(x*2, st, mij, i, val);

    else modificare(x*2+1, mij+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];

    int s1=0, s2=0, mij = (st+dr)/2;

    if(i <= mij) s1 = suma(x*2, st, mij, i, j);

    if(j > mij) s2 = suma(x*2+1, mij+1, dr, i, j);

    return s1+s2;
}

int main()
{
    int n, i, j, x, m, operatie, zi, valoare, inceput, sfarsit;

    f>>n>>m;

    for(i=1;i<=n;i++)
    {
        f>>x;
        creare(1, 1, n, i, x);
    }

//    for(i=1;i<=4*n;i++)
//        cout<<arb[i]<<" ";

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

        if(operatie==0)
        {
            f>>zi>>valoare;
            modificare(1, 1, n, zi, valoare);
        }
        else
        {
            f>>inceput>>sfarsit;
            g<<suma(1, 1, n, inceput, sfarsit)<<endl;
        }
    }
}