Cod sursa(job #1809371)

Utilizator adystar00Bunea Andrei adystar00 Data 18 noiembrie 2016 21:20:42
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <iostream>
#include <fstream>
using namespace std;
int aib[15002],n,m;
int ib(int x)
{
    return x&(-x);
}
void update_plus(int x, int y)
{
    if(x>n)
        return;
    aib[x]+=y;
    x+=ib(x);
    update_plus(x,y);
}
void update (int x, int y)
{
    if(x>n)
        return;
    aib[x]-=y;
    x+=ib(x);
    update(x,y);
}
int query(int x)
{
    int suma=0;
    while(x)
    {
        suma+=aib[x];
        x-=ib(x);
    }
    return suma;
}
int main()
{
    ifstream fin ("datorii.in");
    ofstream fout ("datorii.out");
    int i,x,y,q;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>x;
        update_plus(i,x);
    }
    for(i=1;i<=m;i++)
    {
        fin>>q;
        if(q==0)
        {
            fin>>x>>y;
            update(x,y);
        }
        if(q==1)
        {
            fin>>x>>y;
            fout<<query(y)-query(x-1)<<"\n";
        }
    }
    return 0;
}