Cod sursa(job #1456433)

Utilizator CollermanAndrei Amariei Collerman Data 30 iunie 2015 20:08:14
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.43 kb
#include <fstream>
using namespace std;
ofstream fout("aib.out");
ifstream fin("aib.in");
const int NMAX = 100005;

int n, m, val, op, x, y;
int AIB[NMAX];

int nr_binar(int val)
{
    return val & (-val);
}

void update(int poz)
{
    for(int i = poz; i <= n; i += nr_binar(i))
        AIB[i] += val;
}

int query(int poz)
{
    int suma = 0;

    for(int i = poz; i > 0; i -= nr_binar(i))
        suma += AIB[i];

    return suma;
}

int find_pos(int val)
{
    int st = 1, dr = n, mij = 0;

    while(st <= dr) {
        mij = st + (dr - st) / 2;
        if(query(mij) == val)
            return mij;
        else if(query(mij) < val)
            st = mij + 1;
        else
            dr = mij - 1;
    }

    if(query(mij) == val) return mij;
    return -1;
}

int main()
{
    fin >> n >> m;
    for(int i=1; i<=n; i++) {
        fin >> val;
        update(i);
    }

    for(int i=1; i<=m; i++) {
        fin >> op >> x;
        switch (op) {
            case 0:
                fin >> val;
                update(x);
                break;

            case 1:
                fin >> val;
                fout << query(val) - query(x-1) << '\n';
                break;

            case 2:
                fout << find_pos(x) << '\n';
                break;

            default:
                break;
        }
    }

    fout.close();
    fin.close();
    return 0;
}