Cod sursa(job #2903412)

Utilizator IonescuRalucaIonescu Andreea Raluca IonescuRaluca Data 17 mai 2022 16:00:42
Problema Datorii Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.43 kb
#include <iostream>
#include <fstream>
using namespace std;

int a[60006];
void update(int nod, int st, int dr, int poz, int val)
{
    if(st==dr)
    {
        a[nod]=val;
        return;
    }
    int mij=(st+dr)/2;
    if(poz<=mij)  update(2*nod, st,mij, poz,val);//subarb stang
        else  update(2*nod+1,mij+1,dr,poz,val);//subarb drept

    a[nod]=a[2*nod]+a[2*nod+1]; //suma fiilor
}
void achitare(int nod, int st, int dr, int poz, int val)
{
    if(st==dr)
    {
        a[nod]=a[nod]-val; //se scade din acel nod valoarea achitata
        return;
    }
    int mij=(st+dr)/2;

    if(poz<=mij)   achitare(2*nod, st,mij, poz,val);

        else  achitare(2*nod+1,mij+1,dr,poz,val);

    a[nod]=a[2*nod]+a[2*nod+1];
}
int query(int nod, int st, int dr, int x,int y)
{
    if(x<=st && dr<=y) return a[nod];

    int suma1=0, suma2=0;

    int mij =(st+dr)/2;

    if(x<=mij)  suma1=query(2*nod,st,mij,x,y);

    if(y>mij)  suma2=query(2*nod+1, mij+1,dr, x ,y);

    return suma1+suma2;
}

int main()
{   ifstream f("datorii.in");
    ofstream g("datorii.out");
    int n,m,cod,ziua,val,p,q,x,i;
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        update(1,1,n,i,x);
    }
    for(i=1;i<=m;i++)
    {
        f>>cod;
        if(cod==0)  //achitare
        {   f>>ziua>>val;
            achitare(1,1,n,ziua,val);
        }
        if(cod==1)  //interogare
        {   f>>p>>q;
            g<<query(1,1,n,p,q)<<'\n';
        }
    }
    return 0;
}