Cod sursa(job #2708937)

Utilizator Mihai_EduardMihai Eduard Mihai_Eduard Data 19 februarie 2021 16:31:37
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <iostream>
#include <fstream>
#include <cmath>

#define MX 15001

using namespace std;

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

int n, m, t, a, b;

int v[4*MX], sol;

void build(int nod, int st, int dr)
{
    if(st==dr)
        fin>>v[nod];
    else
    {
        int mij=(st+dr)/2;
        build(2*nod,st,mij);
        build(2*nod+1,mij+1,dr);
        v[nod]=v[2*nod]+v[2*nod+1];
    }
}

void update(int nod, int st, int dr, int poz, int val)
{
    if(st==dr)
        v[nod]=v[nod]-val;
    else
    {
        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);
        v[nod]=v[2*nod]+v[2*nod+1];
    }
}

void query(int nod, int st, int dr, int poz1, int poz2)
{
    if(poz1<=st and poz2>=dr)
    {
        sol+=v[nod];
    }
    else
    {
        int mij=(st+dr)/2;
        if(poz1<=mij)
            query(2*nod,st,mij,poz1,poz2);
        if(poz2>=mij+1)
            query(2*nod+1,mij+1,dr,poz1,poz2);
    }
}

int main()
{
    fin>>n>>m;
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        fin>>t>>a>>b;
        if(t==1)
        {
            sol=0;
            query(1,1,n,a,b);
            fout<<sol<<'\n';
        }
        else
        {
            update(1,1,n,a,b);
        }
    }
    return 0;
}