Cod sursa(job #1498475)

Utilizator refugiatBoni Daniel Stefan refugiat Data 8 octombrie 2015 17:25:29
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.3 kb
#include<fstream>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;
#define len 100002
long long aib[len];
inline void update(long long i,long long s)
{
    for(;i<len;i+=i&(-i))
        aib[i]+=s;
}
inline long long query(long long i)
{
    long long sum=0;
    for(;i>0;i-=i&(-i))
        sum+=aib[i];
    return sum;
}
inline long long caut(long long s,long long n)
{
    long long 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");
    long long n,m;
    si>>n>>m;
    long long i,a;
    for(i=0;i<n;++i)
    {
        si>>a;
        update(i+1,a);
    }
    long long 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;
}