Cod sursa(job #1498474)

Utilizator refugiatBoni Daniel Stefan refugiat Data 8 octombrie 2015 17:24:16
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.22 kb
#include<fstream>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;
#define len 100002
int aib[len];
inline void update(int i,int s)
{
    for(;i<len;i+=i&(-i))
        aib[i]+=s;
}
inline int query(int i)
{
    int sum=0;
    for(;i>0;i-=i&(-i))
        sum+=aib[i];
    return sum;
}
inline int caut(int s,int n)
{
    int poz=0,i;
    for(i=log2(n);i>=0;--i)
    {
        if(aib[poz+(1<<i)]<s)
        {
            poz+=(1<<i);
            s-=aib[poz];
        }
    }
    return poz;
}
int main()
{
    ifstream si;
    si.open("aib.in");
    ofstream so;
    so.open("aib.out");
    int n,m;
    si>>n>>m;
    int i,a;
    for(i=0;i<n;++i)
    {
        si>>a;
        update(i+1,a);
    }
    int b,c;
    for(i=0;i<m;++i)
    {
        si>>a;
        if(a==0)
        {
            si>>b>>c;
            update(b,c);
        }
        else
        {
            if(a==1)
            {
                si>>b>>c;
                so<<query(c)-query(b-1)<<'\n';
            }
            else
            {
                si>>b;
                so<<caut(b,n)+1<<'\n';
            }
        }
    }
    si.close();
    so.close();
    return 0;
}