Cod sursa(job #2501508)

Utilizator andu2006Alexandru Gheorghies andu2006 Data 29 noiembrie 2019 20:04:10
Problema Arbori indexati binar Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.37 kb
#include<bits/stdc++.h>
#define NMAX 100001

using namespace std;

ifstream fin("aib.in");
ofstream fout("aib.out");
int aib[NMAX],n,v[NMAX];
void precalc(){
    for(int i=1;i<=n;i++){
        aib[i]=v[i];
        int p=1,first=(i&(i-1));
        while(p<i-first){
            aib[i]+=aib[i-p];
            p*=2;
        }
    }
}
int p2(int x){
    return ((x^(x-1))&x);
}
void actualizare(int p,int x){
    for(int i=p;i<=n;i=i+p2(i))
        aib[i]+=x;
}
int suma(int p){
    int s=0;
    for(int i=p;i>0;i=i-p2(i))
        s+=aib[i];
    return s;
}
int binsearch(int a){
    int s=1,d=n,m,sum;
    while(s<d){
        m=(s+d)/2; sum=suma(m);
        if(a>sum)
            s=m+1;
        else if(a<sum)
            d=m-1;
        else
            return m;
    }
    if(suma((s+d)/2)==a)
        return (s+d)/2;
    return -1;
}
int main()
{
    ios_base::sync_with_stdio(false);
    int m,i,a,b,querytype;
    fin>>n>>m;
    for(i=1;i<=n;i++){
        fin>>v[i];
    }
    precalc();
    for(i=0;i<m;i++){
        fin>>querytype;
        if(querytype==0){
            fin>>a>>b;
            actualizare(a,b);
        }
        else if(querytype==1){
            fin>>a>>b;
            fout<<suma(b)-suma(a-1)<<'\n';
        }
        else{
            fin>>a;
            fout<<binsearch(a)<<'\n';
        }
    }
    return 0;
}