Cod sursa(job #2549414)

Utilizator PaulOrasanPaul Orasan PaulOrasan Data 17 februarie 2020 17:50:00
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <fstream>

using namespace std;

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

const int NMAX=15010;

int n,m;
int v[NMAX], aib[NMAX];
int s[NMAX];
void citire()
{
    fin>>n>>m;
    for (int i=1; i<=n; i++){
        fin>>v[i];
        s[i]=s[i-1]+v[i];
    }
}
inline int suma(int i,int j)
{
    return s[j]-s[i-1];
}
void construireAIB()
{
    for (int i=1; i<=n; i++)
        aib[i]=suma(i-(i&(~(i-1)))+1,i);     //i-ultimulBit(i)+1
}
int query(int i)
{
    int sum=0;
    while (i>0){
        sum+=aib[i];
        i-=(i&(~(i-1)));
    }
    return sum;
}
void update(int a, int b)
{
    while (a<=n){
        aib[a]-=b;
        a+=(a&(~(a-1)));
    }
}
void rezolvare()
{
    int a,b,c;
    for (int i=1; i<=m; i++){
        fin>>c>>a>>b;
        if (c==0){
            update(a,b);
        }
        if (c==1){
            fout<<query(b)-query(a-1)<<'\n';
        }
    }
}
int main()
{
    citire();
    construireAIB();
    rezolvare();
    fin.close();
    fout.close();
    return 0;
}