Cod sursa(job #3229104)

Utilizator Floroiu_MariusFloroiu Marius Cristian Floroiu_Marius Data 13 mai 2024 20:53:12
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.33 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
vector <int> v(15003);
vector <long long> a;
int n;
void build (int nod,int st,int dr)
{
    if (st==dr)
    {
        a[nod]=v[st];
    }
    else
    {
        int mij=((st+dr)>>1);
        build (2*nod,st,mij);
        build(2*nod+1,mij+1,dr);
        a[nod]=a[2*nod]+a[2*nod+1];
    }
}
int query(int nod,int fiu1,int fiu2,int st,int dr)
{
    if (st>fiu2 || dr<fiu1)
        return 0;
    if (st<=fiu1 && dr>=fiu2)
        return a[nod];
    int mij=((fiu1+fiu2)>>1);
    return query(2*nod,fiu1,mij,st,dr)+query(2*nod+1,mij+1,fiu2,st,dr);
}
void update(int nod,int st,int dr,int indice,int val)
{
    if (st==dr)
    {
        a[nod]-=val;
        return ;
    }
    int mij=((st+dr)>>1);
    if (indice<=mij)
        update(2*nod,st,mij,indice,val);
    else
        update(2*nod+1,mij+1,dr,indice,val);
    a[nod]=a[2*nod]+a[2*nod+1];
}
int c,m,x,y;
int main()
{
    fin>>n>>m;
    a.resize(4*n);
    for (int i=0; i<n; i++)
        fin>>v[i];
    build(1,0,n-1);
    while (m--)
    {
        fin>>c>>x>>y;
        if (c==1)
        {
            fout<<query(1,0,n-1,x-1,y-1)<<'\n';
        }
        else
        {
            update(1,0,n-1,x-1,y);
        }
    }
    return 0;
}