Cod sursa(job #2666845)

Utilizator alexandrubunea03Bunea Alexandru alexandrubunea03 Data 2 noiembrie 2020 15:53:34
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.71 kb
#include <bits/stdc++.h>
using namespace std;
#define debug_var(x) cout << #x << '=' << x << endl;
#define debug_stl(x) for(auto k : x) cout << k << ' '; cout << endl;
#define debug_map(x) for(auto k : x) cout << k.first << ':' << k.second << endl;
#define read_vector(x, s, k) for(auto i = s; i <= k; ++i) cin >> x[i];
#define read_2dvector(x, s, l1, l2) for(auto i = s; i <= l1; ++i) for(auto j = s; j <= l2; ++j) cin >> x[i][j];
#define vec vector
#define ull unsigned long long int
#define ll long long int
#define str string
#define umap unordered_map
#define uset unordered_set
#define mset multiset
#define umset unordered_multiset
#define NMax 100000

int aib[NMax + 1], n, m;

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

int main() {
  freopen("aib.in", "r", stdin);
  freopen("aib.out", "w", stdout);
  ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
  cin >> n >> m;
  for(int i = 1; i <= n; ++i) {
    int x;
    cin >> x;
    add(i, x);
  }
  for(int i = 1; i <= m; ++i) {
    int type, x, y;
    cin >> type;
    switch (type) {
      case 0: {
        cin >> x >> y;
        add(x, y);
        break;
      }
      case 1: {
        cin >> x >> y;
        cout << sum(y) - sum(x - 1) << '\n';
        break;
      }
      case 2: {
        cin >> x;
        int i, step;
        for(step = 1; step <= n; step <<= 1);
        for(i = 0; step; step >>= 1)
          if(step + i <= n && aib[step + i] <= x) i += step;
        if(!i) i = -1;
        cout << i << '\n';
        break;
      }
    }
  }
  return 0;
}