Cod sursa(job #3256396)

Utilizator MilitaruMihaiMihaiMIlitaru MilitaruMihai Data 14 noiembrie 2024 15:05:44
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.12 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n,t,v[100005],aib[100005];
void adun(int x,int s){
    for (int i=x;i<=n;i+=i&(-i))
        aib[i]+=s;
}
int suma(int x){
    int s=0;
    for (int i=x;i>=1; i-=i&(-i))
        s+=aib[i];
    return s;
}
int main()
{
    fin>>n>>t;
    for (int i=1;i<=n;i++){
        fin>>v[i];
        adun(i,v[i]);
    }
    for (;t;--t){
        int op;
        fin>>op;
        if (op==0) {
            int x,p;
            fin>>x>>p;
            adun(x,p);
        }
        else if (op==1) {
            int a,b;
            fin>>a>>b;
            fout<<suma(b)-suma(a-1)<<'\n';
        }
        else if (op==2) {
            int k;
            fin>>k;
            int st=1,dr=n,mij,sol=-1;
            while (st<=dr){
                mij=(st+dr)/2;
                int s=suma(mij);
                if (s==k) sol=mij,st=dr+1;
                    else if (s<k) st=mij+1;
                         else dr=mij-1;
            }
            fout<<sol<<'\n';
        }
    }
    return 0;
}