Pagini recente » Cod sursa (job #949748) | Cod sursa (job #996028) | Cod sursa (job #2171787) | Cod sursa (job #1168083) | Cod sursa (job #2175642)
#include <bits/stdc++.h>
using namespace std;
#define LSB(x) ((-x) & x)
template<typename T = int>
struct BIT {
// UPDATE UPDATE
// PREFIX_SUM SUM[1, ..., k]
// QUERY SUM[i, j]
// FIND_K SUM[1...k] == S
int size;
vector<T> bit; // bit[0], bit[1], ..., bit[size]
// because size is fixed, I can precompute STEP_MAX for FIND_K
T STEP_MAX;
BIT(int _size) : size(_size), bit(size + 1, 0) {
// precompute max_step
STEP_MAX = 1;
while (STEP_MAX < size) {
STEP_MAX <<= 1;
}
}
void update(int k, const T& x) {
for(; k <= size; k += LSB(k)) {
bit[k] += x;
}
}
int prefix_sum(int k) {
T sum = 0;
for(; k > 0; k -= LSB(k)) {
sum += bit[k];
}
return sum;
}
T query(int i, int j) {
return prefix_sum(j) - prefix_sum(i - 1);
}
int find_k(T sum) {
int k = 0;
for (int step(STEP_MAX); step > 0; step >>= 1) {
if (k + step <= size && bit[k + step] <= sum) {
k += step;
sum -= bit[k];
if (sum == 0) {
return k;
}
}
}
return -1;
}
};
int main() {
std::ios::sync_with_stdio(false);
ifstream fin("aib.in");
ofstream fout("aib.out");
int n, m;
fin >> n >> m;
BIT<long long> bit(n);
for (int i = 1; i <= n; ++i) {
int x;
fin >> x;
// update
bit.update(i, x);
}
while (m--) {
int type, start, end, x, pos;
fin >> type;
switch (type) {
case 0: {
fin >> pos >> x;
bit.update(pos, x);
break;
}
case 1: {
fin >> start >> end;
fout << bit.query(start, end) << "\n";
break;
}
case 2: {
fin >> x;
fout << bit.find_k(x) << "\n";
break;
}
default: {
fout << "wtf?";
break;
}
}
}
fin.close();
fout.close();
return 0;
}