Cod sursa(job #2492325)

Utilizator AlexPascu007Pascu Ionut Alexandru AlexPascu007 Data 14 noiembrie 2019 15:52:52
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <bits/stdc++.h>
#define DIM 100010
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n,m,x,i,a,b,sol,cod,A[DIM];
void update(int poz, int val) { //updatez A[poz]
    for (;poz<=n;poz+=(poz&-poz))
        A[poz]+=val;
}
int query(int poz) { //returneaza suma A[1]+...A[poz]
    int val=0;
    for (;poz!=0;poz-=(poz&-poz))
        val+=A[poz];
    return val;
}
int cb(int val) { //returneaza pozitia pentru care suma A[1]+...+A[poz]=val
    int st=1, dr=n;
    while (st<=dr) {
        int mid=(st+dr)/2;
        if (query(mid)==val)
            return mid;
        if (query(mid)>val)
            dr=mid-1;
        else
            st=mid+1;
    }
    return -1;
}
int main() {
    fin>>n>>m;
    for (i=1;i<=n;i++) {
        fin>>x;
        update(i,x); //initializare
    }
    while (m--) {
        fin>>cod;
        switch (cod) {
            case 0: {
                fin>>a>>b;
                update(a,b);
                break;
            }
            case 1: {
                fin>>a>>b;
                fout<<query(b)-query(a-1)<<"\n";
                break;
            }
            case 2: {
                fin>>a;
                fout<<cb(a)<<"\n";
                break;
            }
        }
    }
    return 0;
}