Cod sursa(job #3247792)

Utilizator georgiemanea georege caharisitan georgie Data 9 octombrie 2024 10:09:44
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <fstream>
#include <bits/stdc++.h>
using namespace std;
int n,m;
int v[100001],aib[100001];
int presum(int i){
    int sum=0;
    while(i>0){
        sum+=aib[i];
        i-=(i&-i);
    }
    return sum;
}
int query(int i,int j){
    return presum(j)-presum(i-1);
}
void update(int i,int val){
    while(i<=n){
        aib[i]+=val;
        i+=(i&-i);
    }
}
ifstream fin("aib.in");
ofstream fout("aib.out");
int main()
{
    fin >> n >> m;
    for(int i=1;i<=n;i++){
       fin >> v[i];
       update(i,v[i]);
    }
    for(int i=0;i<m;i++){
        int q,x,y;
        fin >> q >> x;
        if(q==0){
            fin >> y;
            update(x,y);
        }
        else if(q==1){
            fin >> y;
            fout << query(x,y)<< '\n';
        }
        else{
            int st,dr;
            st=1;
            dr=n;
            while(dr>st){
                int mij=(st+dr)/2;
                int a=presum(mij);
                if(a>=x)
                    dr=mij;
                else
                    st=mij+1;
            }
            if(presum(dr)==x)
                fout<< dr << '\n';
            else
                fout << "-1\n";
        }
    }
    return 0;
}