Cod sursa(job #1893337)

Utilizator gabrielamoldovanMoldovan Gabriela gabrielamoldovan Data 25 februarie 2017 17:04:33
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include <iostream>
#include <fstream>

#define zeros(x) ((x^(x-1))&x)
#define nmax 100000

using namespace std;

ifstream f("aib.in");
ofstream g("aib.out");

int aib[nmax], n;

void Add(int x, int val)
{
    for(int i=x; i<=n; i+=zeros(i))
    {
        aib[i]+=val;
    }
}

int Compute(int x)
{
    int sum=0;
    for(int i=x; i>=1; i-=zeros(i))
    {
        sum+=aib[i];
    }
    return sum;
}

int main()
{
    int m;
    int p, a, b;
    f>>n>>m;
    for(int i=1; i<=n; ++i)
    {
        f>>a;
        Add(i, a);
    }
    for(int i=1; i<=m; ++i)
    {
        f>>p;
        if(p==0)
        {
            f>>a>>b;
            Add(a, b);
        }
        if(p==1)
        {
            f>>a>>b;
            g<<Compute(b)-Compute(a-1)<<"\n";
        }
        if(p==2)
        {
            f>>a;
            /*b=1;
            while(Compute(b)!=a)
            {
                ++b;
            }
            g<<b<<"\n";*/
        }
    }
    return 0;
}