Cod sursa(job #2421798)

Utilizator RadulescuRichardAndreiRadulescu Richard- Andrei RadulescuRichardAndrei Data 16 mai 2019 10:28:04
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <iostream>
#include <fstream>
#define NMAX 15005
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");


int aint[4*NMAX];
int pret[NMAX];
int n,m;

int build(int nod,int st,int dr);
int Suma(int nod,int st,int dr,int a, int b);
void update(int nod,int st,int dr,int val,int poz);
int main()
{
    int op,a,b,i;
    fin>>n>>m;
    for(i=1;i<=n;i++)
        fin>>pret[i];
    build(1,1,n);
    for(i=1;i<=m;i++)
    {
        fin>>op>>a>>b;
        if(op==1)
            fout<<Suma(1,1,n,a,b)<<'\n';
            else
            update(1,1,n,a,b);
    }

    return 0;
}

int build(int nod,int st,int dr)
{
    int s=0;
    if(st==dr)
    {
        aint[nod]=pret[st];
        return aint[nod];
    }
    int mij=(st+dr)/2;
   s+= build(2*nod,st,mij);
    s+=build(2*nod+1,mij+1,dr);
    aint[nod]=s;
    return s;
}

int Suma(int nod,int st,int dr,int a, int b)
{
    int s=0;
    if(st>=a && dr<=b)
        return aint[nod];
    if(dr<a || b<st ) return 0;

    int mij=(st+dr)/2;
    s+=Suma(s*nod,st,mij,a,b);
    s+=Suma(s*nod+1,mij+1,dr,a,b);
    return s;

}

void update(int nod,int st,int dr,int val,int poz)
{
if(st==dr)
    {
     aint[nod]-=val;
     return;
    }

int mij=(st+dr)/2;
if(poz<=mij)
    update(nod*2,st,mij,val,poz);
        else
    update(nod*2+1,mij+1,dr,val,poz);

aint[nod]=aint[nod*2]+aint[nod*2+1];
}