Cod sursa(job #2678848)

Utilizator KPP17Popescu Paul KPP17 Data 28 noiembrie 2020 20:31:53
Problema Arbori indexati binar Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <fstream>
#define fisier "aib"
std::ifstream in(fisier ".in");
std::ofstream out(fisier ".out");
int V[100001], n, m;
constexpr int S(int i)
{return i? V[i] + S(i ^ i & -i): 0;}
inline void U(int i, int e)
{if (i <= n) V[i] += e, U(i + (i & -i), e);}
inline int P(int s)
{
    int i = 1, j = n, m;
    while (i <= j)
        if (s < S(m = (i+j)>>1))
            j = m - 1;
        else i = m + 1;
    return S(j) == s? j: -1;
}
int main()
{
    std::ios::sync_with_stdio(false);
    in >> n >> m;
    for (int i = 1; i <= n; i++)
        {int e; in >> e; U(i, e);}
    while (m--)
    {
        int v, a, b; in >> v >> a; if (v < 2) in >> b;
        switch (v)
        {
            case 0: U(a, b); break;
            case 1: out << S(b) - S(a-1) << '\n'; break;
            case 2: out << P(a) << '\n';
        }
    }
}