Cod sursa(job #2800739)

Utilizator UnknownPercentageBuca Mihnea-Vicentiu UnknownPercentage Data 13 noiembrie 2021 21:03:31
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.64 kb
#include <bits/stdc++.h>

using namespace std;

inline char gc() { // like getchar()
    static char buf[1 << 16];
    static size_t bc, be;
    if (bc >= be) {
        buf[0] = 0, bc = 0;
        be = fread(buf, 1, sizeof(buf), stdin);
    }
    return buf[bc++]; // returns 0 on EOF
}
 
int readInt() {
    int a, c;
    while ((a = gc()) < 40);
    if (a == '-') return -readInt();
    while ((c = gc()) >= 48) a = a * 10 + c - 480;
    return a - 48;
}
 
// from madhur4127
/** Write */
static const int buf_size = 1 << 16;
static int write_pos = 0;
static char write_buf[buf_size];
uint32_t digits10(uint64_t v) {
    int len = 1;
    while (v >= 10000)
        v /= 10000, len += 4;
    return len + (v < 100 ? v >= 10 : 2 + (v >= 1000));
}
unsigned u64ToAsciiTable(uint64_t value) {
    static const char digits[201] = "00010203040506070809101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899";
    uint32_t const length = digits10(value);
    uint32_t next = length - 1;
    char *dst = &write_buf[write_pos];
    for (; value >= 100; next -= 2, value /= 100) {
        auto const i = (value % 100) * 2;
        dst[next] = digits[i + 1], dst[next - 1] = digits[i];
    }
    // Handle last 1-2 digits
    if (value < 10) {
        dst[next] = '0' + value;
    } else {
        auto i = uint32_t(value) * 2;
        dst[next] = digits[i + 1], dst[next - 1] = digits[i];
    }
    return length;
}
template <typename T = int> inline void write(T x, char end = 0) {
    if (__builtin_expect(write_pos + 21 >= buf_size, 0)) fwrite(write_buf, 1, write_pos, stdout), write_pos = 0;
    if (x < 0) write<char>('-'), x = -x;
    write_pos += u64ToAsciiTable(x);
    if (end) write<char>(end);
}
template <> inline void write<char>(const char x, char) {
    if (__builtin_expect(write_pos == buf_size, 0)) fwrite(write_buf, 1, buf_size, stdout), write_pos = 0;
    write_buf[write_pos++] = x;
}
template <> inline void write<string>(const string s, char end) {
    for (const char c : s)
        write<char>(c);
    if (end) write<char>(end);
}
struct Flusher {
    ~Flusher() {
        if (write_pos) fwrite(write_buf, 1, write_pos, stdout), write_pos = 0;
    }
} flusher;

inline void Open(const string Name) {
    #ifndef ONLINE_JUDGE
        (void)!freopen((Name + ".in").c_str(), "r", stdin);
        (void)!freopen((Name + ".out").c_str(), "w", stdout);
    #endif
}

int bit[100001];
int N, M;

void add(int idx, int plus) {
    for(;idx <= N;idx += idx & -idx)
        bit[idx] += plus;
}

int sum(int idx) {
    int res = 0;
    for(;idx > 0;idx -= idx & -idx)
        res += bit[idx];

    return res;
}

int Query(int L, int R) {
    return sum(R) - sum(L - 1); 
}

int Find(int x) {
    int y = 0;
    for(int k = (1 << 17);k >= 1;k >>= 1) {
        if(y + k <= N && bit[y + k] <= x) {
            x -= bit[y + k], y += k;
            if(x == 0) 
                return y;
        }
    }

    return -1;
}

int main() {

    Open("aib");

    N = readInt(), M = readInt();
    for(int i = 1, plus;i <= N;i++)
        plus = readInt(), add(i, plus);

    while(M--) {
        int T, a, b;
        T = readInt();
        if(T == 0)
            a = readInt(), b = readInt(), add(a, b);
        if(T == 1) {
            a = readInt(), b = readInt();
            write(Query(a, b));
            write('\n');
        }
        if(T == 2) {
            a = readInt();
            write(Find(a));
            write('\n');
        }
    }

    return 0;
}