Cod sursa(job #2289296)

Utilizator 3DwArDPauliuc Edward 3DwArD Data 24 noiembrie 2018 12:45:48
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include<bits/stdc++.h>
#define NMAX 100010
using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");
int n,m,A[NMAX];;
void update(int poz, int val){
    while(poz<=n){
            A[poz]+=val;
            poz+=(poz&(-poz));
    }
}
int query(int poz){
    long long sum=0;
    while(poz){
        sum+=A[poz];
        poz-=(poz&(-poz));
    }
    return sum;
}
void citire(){
    f>>n>>m;
    for(int i=1,x;i<=n;i++){
        f>>x;
        update(i,x);
    }
}
int caut_bin(int k){
    int l=1,r=n,val,mid;
    while(l<=r){
        mid=(l+r)/2;
        val=query(mid);

        if(val==k)return mid;
        if(val<k)l=mid+1;
        if(val>k)r=mid-1;
    }
    return -1;
}
int main()
{
    citire();
    for(int i=1,t,x,y;i<=m;i++){
        f>>t;
        if(t==0){
            f>>x>>y;
            update(x,y);
        }
        if(t==1){
            f>>x>>y;
            g<<query(y)-query(x-1)<<"\n";
        }
        if(t==2){
            f>>x;
            g<<caut_bin(x)<<"\n";
        }
    }
    return 0;
}