Cod sursa(job #1550015)

Utilizator herbertoHerbert Mohanu herberto Data 13 decembrie 2015 01:26:07
Problema Arbori indexati binar Scor 100
Compilator c Status done
Runda Arhiva educationala Marime 1.1 kb
#include <stdio.h>
#include <stdlib.h>
#define MAXN 150004

int v[MAXN];
int x, y, n;
int lsb(int x){
  return x^(x&(x-1));
}
void update(int x, int y){
  while(x<=n){
    v[x]+=y;
    x+=lsb(x);
  }
}
inline int querry(int x){
  int s;
  s=0;
  while(x>0){
    s+=v[x];
    x-=lsb(x);
  }
  return s;
}

int cautbin(int st, int dr, int h){
  int mij;
  while(dr-st>1){
    mij=(dr+st)/2;
    if(h<=querry(mij))
      dr=mij;
    else
      st=mij;
  }
  if(querry(dr)==h && dr<=n)
    return dr;

  return -1;
}

int main(){
  FILE*fin=fopen("aib.in", "r");
  FILE*fout=fopen("aib.out", "w");
  int m, tip, p, i, nr;
  fscanf(fin, "%d%d", &n, &m);
  for(i=1; i<=n; i++){
    fscanf(fin, "%d", &nr);
    update(i, nr);
  }

  for(i=1; i<=m; i++){
    fscanf(fin, "%d", &tip);
    if(tip!=2){
      fscanf(fin, "%d%d", &x, &y);
      if(tip==0)
        update(x, y);

      if(tip==1)
        fprintf(fout, "%d\n", querry(y)-querry(x-1));
    }
    else{
      fscanf(fin, "%d", &x);
      fprintf(fout, "%d\n", cautbin(0, n+1, x));
    }
  }
  return 0;
}