Cod sursa(job #1473875)

Utilizator Vlad_lsc2008Lungu Vlad Vlad_lsc2008 Data 20 august 2015 13:38:35
Problema Arbori indexati binar Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.07 kb
#include <iostream>
#include <cstdio>
#define fin "aib.in"
#define fou "aib.out"
#define Max 100010
using namespace std;
int n,m,aib[Max];

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

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

int query(int poz)
{
    int i,rez=0;
    for(i=poz;i>=1;i-=zeros(i)) rez+=aib[i];
    return rez;
}

int getpos(int val)
{
    int sum=0,idx=1;
    while(sum<val)
    {
        sum=query(idx);
        if(sum==val) return idx;
        idx++;
    }
    return -1;
}

int main()
{
    int i,x;
    freopen(fin,"r",stdin);
    freopen(fou,"w",stdout);
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&x);
        update(i,x);
    }
    int cod,a,b;
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&cod,&a);
        if(cod!=2) scanf("%d",&b);
        if(cod==0) update(a,b);
        if(cod==1) printf("%d\n",query(b)-query(a-1));
        if(cod==2) printf("%d\n",getpos(a));
    }
    fclose(stdout);
    fclose(stdin);
    return 0;
}