Cod sursa(job #1794271)

Utilizator alexsandulescuSandulescu Alexandru alexsandulescu Data 1 noiembrie 2016 09:37:35
Problema Arbori indexati binar Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include <fstream>
#define ub(n) (n & (-n))

using namespace std;

ifstream f ("aib.in");
ofstream g ("aib.out");

int N, M, c, x, j, a, b, SS;
int AIB[100003];

int add(int poz, int val) {
    for(int i = poz; i <= N; i += ub(i)) AIB[i] += val;
    return 0;
}
int sum(int poz) {
    int sum = 0;
    for(int i = poz; i >= 1; i -= ub(i)) sum += AIB[i];
    return sum;
}

int main()
{
    f >> N >> M;
    for(int i = 1; i <= N; i++)
        f >> x, add(i, x);
    for(int i = 1; i <= M; i++) {
        f >> c;
        if(c == 0) f >> j >> x, add(j, x);
        else if(c == 2) {
            f >> x, j = 1, SS = AIB[1]; /// AI GRIJA! POATE TREBUIE SA INITIALIZEZI ALTCUMVA DECAT DE OBICEI!!! >:(
            while(sum(j) < x) j++, SS = sum(j);
            if(SS != x) g << -1 << "\n";
            else        g << j << "\n";
        }
        else if (c == 1) f >> a >> b, g << sum(b) - sum(a-1) << "\n";
    }
    return 0;
}