Cod sursa(job #2646546)

Utilizator BogdanRazvanBogdan Razvan BogdanRazvan Data 1 septembrie 2020 14:26:53
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.4 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin ("aib.in");
ofstream fout ("aib.out");

const int DIM = 100000 + 5;

int a[DIM], fen[DIM];

void add(int node, int val)
{
    for(; node <= DIM - 5; node += node & (-node)) {
        fen[node] += val;
    }
}

int sum(int node)
{
    int sol = 0;
    for(; node > 0; node -= node & (-node)) {
        sol += fen[node];
    }
    return sol;
}

int main()
{
    int n, m;

    fin >> n >> m;
    for(int i = 1; i <= n; ++i) {
        fin >> a[i];
        add(i, a[i]);
    }
    for(int i = 1; i <= m; ++i) {
        int type, a, b;

        fin >> type;
        if(type == 0) {
            fin >> a >> b;
            add(a, b);
        }
        if(type == 1) {
            fin >> a >> b;
            fout << sum(b) - sum(a - 1) << "\n";
        }
        else if(type == 2) {
            fin >> a;
            int left = 1, right = n;
            while(left <= right) {
                int mid = (left + right) >> 1;
                if(sum(mid) > a) {
                    right = mid - 1;
                }
                else if(sum(mid) < a) {
                    left = mid + 1;
                }
                else {
                    fout << mid << "\n";
                    break;
                }
            }
            if(left > right) fout << "-1" << "\n";
        }
    }
    return 0;
}