Cod sursa(job #2882926)

Utilizator bubblegumixUdrea Robert bubblegumix Data 31 martie 2022 22:26:14
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.1 kb
#include<bits/stdc++.h>
#define all(v) v.begin(), v.end()
#define lsb(x) ((x) & (-x))
#define hmap unordered_map
#define var auto &
#define hset unordered_set
#define pq priority_queue
#define exists(x, v) (v.find(x) != v.end())
#define inrange(x, a, b) (x >= a && x <= b)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long int ull;
typedef pair<int, int> p32;
typedef pair<ll, ll> p64;
typedef pair<double, double> pdd;
typedef pair<ull, ull> p64_u;
typedef vector<int> v32;
typedef vector<ll> v64;
typedef vector<double> v64_d;
typedef vector<ull> v64_u;
typedef vector<p32> vp32;
typedef vector<p64> vp64;
typedef vector<p64_u> vp64_u;
typedef vector<pdd> vpdd;
typedef vector<string> vstr;
typedef vector<vector<int>>  vv32;
typedef vector<v64> vv64;
#define sim template < class c
#define ris return * this
#define dor > debug & operator <<
#define eni(x) sim > typename \
  enable_if<sizeof dud<c>(0) x 1, debug&>::type operator<<(c i) {
#define sim template < class c
#define ris return * this
#define dor > debug & operator <<
#define eni(x) sim > typename \
  enable_if<sizeof dud<c>(0) x 1, debug&>::type operator<<(c i) {
sim > struct rge { c b, e; };
sim > rge<c> range(c i, c j) { return rge<c>{i, j}; }
sim > auto dud(c* x) -> decltype(cerr << *x, 0);
sim > char dud(...);
struct debug {
#ifdef HOME
    ~debug() { cerr << endl; }
    eni(!= ) cerr << boolalpha << i; ris;
}
eni(== ) ris << range(begin(i), end(i));
}
sim, class b dor(pair < b, c > d) {
    ris << "(" << d.first << ", " << d.second << ")";
}
sim dor(rge<c> d) {
    *this << "[";
    for (auto it = d.b; it != d.e; ++it)
        *this << ", " + 2 * (it == d.b) << *it;
    ris << "]";

}


#else
    sim dor(const c&) { ris; }
#endif
};
#define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "
#define darr(x,n) debug() <<#x<<": "<< range(x,x+n+1)
#define dmat(x,n,m) for(int i=0;i<=n;i++) if(!i) debug() <<#x << ":" << range(x[i],x[i]+m+1); else debug() << range(x[i],x[i]+m+1);

const ll lim = 1e5 + 5;
ll aib[lim], a[lim];
int n, m;

void add(int pos, ll val)
{
    for (int i = pos; i <= n; i += lsb(i))
        aib[i] += val;
}
ll sum(int pos)
{
    ll s = 0;
    for (int i = pos; i > 0; i -= lsb(i))
        s += aib[i];
    return s;
}

int bs(ll val)
{
    int l = 1, r = n;
    while (l <= r)
    {
        int m = (l + r) >> 1;
        if (sum(m) >= val)
            r = m - 1;
        else
            l = m + 1;

    }
    return l;
}

int main()
{
    // #ifdef HOME
    freopen("aib.in", "r", stdin);
    freopen("aib.out", "w", stdout);
    // #endif
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        add(i, a[i]);
    }
    while (m--)
    {
        int op, x, y;
        cin >> op >> x;
        darr(aib, n);
        if (op == 0)
        {
            cin >> y;
            add(x, y);
        }
        else
            if (op == 1)
            {
                cin >> y;
                cout << sum(y) - sum(x - 1) << "\n";
            }
            else
                cout << bs(x) << "\n";
    }
    return 0;
}