Cod sursa(job #583750)

Utilizator stef93Stefan Gilca stef93 Data 22 aprilie 2011 14:37:52
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.97 kb
#include <fstream>
using namespace std;
ifstream in("aib.in");
ofstream out("aib.out");

int aib[100003],n,m;

int bit(int x)
{
    return (x&(x-1))^x;
}

void update(int x,int z)
{
    int i;
    for(i=x;i<=n;i+=bit(i))
    aib[i]+=z;
}

int sum(int x)
{
    int i,suma=0;
    for(i=x;i>0;i-=bit(i))
    suma+=aib[i];
    return suma;
}

int minp(int x)
{
    int step=1,i;
    for(;step<=n;step<<=1);
    for(i=n;step;step>>=1)
    if(i-step>0&&sum(i-step)>=x)i-=step;
    return i;
}

int main()
{
    int i,x,op,y;
    in>>n>>m;
    for(i=1;i<=n;i++)
    in>>x,update(i,x);
    for(;m;--m)
    {
        in>>op;
        if(op==0)
        {
            in>>x>>y;
            update(x,y);
        }
        if(op==1)
        {
            in>>x>>y;
            out<<sum(y)-sum(x-1)<<'\n';
        }
        if(op==2)
        {
            in>>x;
            out<<minp(x)<<'\n';
        }
    }in.close(),out.close();
    return 0;
}