Pagini recente » Cod sursa (job #55037) | Cod sursa (job #3335945) | Cod sursa (job #3335396) | Cod sursa (job #689779) | Cod sursa (job #3356789)
#include <fstream>
#include <iostream>
#include <vector>
using namespace std;
void actualizare(vector <int> &aib, int p, int val) {
int n = (int)aib.size() - 1;
while (p <= n) {
aib[p] += val;
int putere = (p & (-p));
p += putere;
}
}
int interogare(vector <int> &aib, int p) {
int n = (int)aib.size() - 1, suma = 0;
while (p > 0) {
suma += aib[p];
int putere = (p & (-p));
p -= putere;
}
return suma;
}
int interogare_poz_min(vector <int> &aib, int s) {
// cautam (binar) cea mai mica pozitie pe care avem interogare >= s
int n = (int)aib.size() - 1;
int st = 1, dr = n, rez = n + 1;
while (st <= dr) {
int m = (st + dr) / 2;
if (interogare(aib, m) >= s) {
rez = m;
dr = m - 1;
} else {
st = m + 1;
}
}
if (rez > n || interogare(aib, rez) > s) {
rez = -1;
}
return rez;
}
int main() {
ifstream in("aib.in");
ofstream out("aib.out");
int n, nr_op;
in >> n >> nr_op;
vector <int> aib(n + 1, 0);
for (int i = 1; i <= n; i++) {
int val;
in >> val;
actualizare(aib, i, val);
}
for (int i = 0; i < nr_op; i++) {
int tip;
in >> tip;
if (tip == 0) {
int poz, val;
in >> poz >> val;
actualizare(aib, poz, val);
} else if (tip == 1) {
int a, b;
in >> a >> b;
out << interogare(aib, b) - interogare(aib, a - 1) << "\n";
} else {
int s;
in >> s;
out << interogare_poz_min(aib, s) << "\n";
}
}
in.close();
out.close();
return 0;
}