Cod sursa(job #2502219)

Utilizator Bogdan_BuzatuBuzatu Bogdan Mihai Bogdan_Buzatu Data 30 noiembrie 2019 14:56:08
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.21 kb
#include <fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n,nrtst,t,v[100010],x,id,cap1,cap2;
void add(int poz, int val){
    while(poz<=n){
        v[poz]+=val;
        poz+=(poz&-poz);
    }

}
int sum(int poz){
    int s=0;
    while(poz>0){
        s+=v[poz];
        poz-=(poz&-poz);
    }
    return s;
}

int cautbinar(int val){
    int st=0;
    int dr=n+1;
    while(st<=dr){
        int mid=(st+dr)/2;
        if(sum(mid)<val){
            st=mid+1;
        }
        else{
            dr=mid-1;
        }

    }
    if(sum(st)==val){
        return st;
    }

    return -1;


}

int main(){

    fin>>n>>nrtst;
    for(int i=1;i<=n;i++){
        fin>>x;
        add(i,x);
    }


    for(int i=1;i<=nrtst;i++){
        fin>>t;
        if(t==0){
            fin>>id>>x;
            add(id,x);
        }
        if(t==1){
            fin>>cap1>>cap2;
            fout<<sum(cap2)-sum(cap1-1)<<"\n";
        }

        if(t==2){
            fin>>x;
            if(x==0){
                fout<<"-1"<<"\n";
            }
            else{
                fout<<cautbinar(x)<<"\n";
            }
        }


    }


}