Cod sursa(job #1264865)

Utilizator bogdi1bogdan bancuta bogdi1 Data 16 noiembrie 2014 13:47:13
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.27 kb
#include <cstdio>
int n;
int aib[100001];
int v[100001];
int query(int poz)
{
    int s=0;
    for( ; poz; poz-=poz&(poz-1)^poz)
        s=s+aib[poz];
    return s;
}
void update(int poz, int val)
  {
      for( ; poz<=n; poz+=poz&(poz-1)^poz)
        aib[poz]=aib[poz]+val;
  }
using namespace std;

int main()
{   freopen("aib.in", "r",stdin);
    freopen("aib.out", "w",stdout);
    int m,s,i,x,op,a,b,j,h;
    scanf("%d%d", &n, &m);
    for(i=1; i<=n; i++){
        scanf("%d", &x);
        update(i,x);
        v[i]=x;
    }
    for(i=0; i<m; i++){
        scanf("%d", &op);
        if(op==0){
            scanf("%d%d", &a, &b);
            update(a,b);
        }
        if(op==1){
            scanf("%d%d", &a, &b);
            printf("%d\n", query(b)-query(a-1));
        }
        if(op==2){
            scanf("%d", &a);
           for(h=1; h<n; h<<=1);
            for(j=0; h!=0; h=h/2){
                if(h+j<=n){
                    if(a>=aib[h+j]){
                        j=j+h;
                        a=a-aib[h];
                        if(a==0)
                            printf("%d\n", h);
                    }
                }
            }
           if(a)
                printf("-1");
        }
    }
    return 0;
}