Cod sursa(job #1165190)

Utilizator ThomasFMI Suditu Thomas Thomas Data 2 aprilie 2014 15:47:33
Problema Arbori indexati binar Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 0.97 kb
#include <fstream>
using namespace std;

#define NMax 100005

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

int n,m;
int A[NMax];
int LSB[NMax];

void Preproc()
{
    for(int i=1;i<=n;i++) LSB[i]=i&(-i);
}

void update(int poz,int val)
{
    while(poz<=n)
    {
        A[poz]+=val;
        poz+=LSB[poz];
    }
}

int query(int poz)
{
    int sum=0;
    while(poz>0)
    {
        sum+=A[poz];
        poz-=LSB[poz];
    }
    return sum;
}

int bs(int val,int st,int dr)
{
    if(st>dr) return -1;

    int mij=(st+dr)/2;
    int s=query(mij);

    if(s==val) return mij;
    else if(s>val) return bs(val,st,mij-1);
    else return bs(val,mij+1,dr);
}

int main()
{
    int i,a,b,c;

    f>>n>>m;

    Preproc();

    for(i=1;i<=n;i++) f>>a,update(i,a);

    for(i=1;i<=m;i++)
    {
        f>>a;
        if(a==0) f>>b>>c,update(b,c);
        else if(a==1) f>>b>>c,g<<query(c)-query(b-1)<<"\n";
        else f>>b,g<<bs(b,1,n)<<"\n";
    }

    f.close();
    g.close();
    return 0;
}