Pagini recente » Aliens | Mozaic | Cod sursa (job #126918) | Anamaria | Cod sursa (job #2482866)
#include <fstream>
//#include <iostream>
using namespace std;
ifstream cin("aib.in");
ofstream cout("aib.out");
struct aint {
aint *a, *b;
int l, r;
int v;
aint(int y, int z) {
l = y;
r = z;
v = 0;
a = NULL;
b = NULL;
}
aint(){}
};
void build(aint *node) {
int l = node->l;
int r = node->r;
if(l != r) {
node->a = new aint(l, l + (r - l + 1)/2 - 1);
node->b = new aint(l + (r - l + 1)/2, r);
build(node->a);
build(node->b);
}
}
int query(aint *node, int l, int r) {
if((l == node->l) && (r == node->r))
return node->v;
else if(r <= node->a->r)
return query(node->a, l, r);
else if(l >= node->b->l)
return query(node->b, l, r);
else return query(node->a, l, node->a->r) + query(node->b, node->b->l, r);
}
void add(aint *node, int add, int target){
while(true){
node->v += add;
if(node->a == NULL)
return;
if(node->a->r >= target)
node = node->a;
else node = node->b;
}
}
int main() {
int n, m;
cin>>n>>m;
aint root = aint(1, n);
build(&root);
for(int x = 1, tmp; x<=n; x++){
cin>>tmp;
add(&root, tmp, x);
}
for(int x = 0, req;x<m;x++){
cin>>req;
if(req == 0){
int target, add_;
cin>>target>>add_;
add(&root, add_, target);
}else if(req == 1){
int l, r;
cin>>l>>r;
cout<<query(&root, l, r)<<'\n';
}else{
int v;
cin>>v;
int l = 1, r = n, m;
while(l != r){
m = l + (r - l)/2 + 1;
int v1 = query(&root, 1, m);
if(v1 == v){
l = m;
r = m;
break;
}else if(v1 < v){
l = m + 1;
}else r = m - 1;
}
m = l;
if(query(&root, 1, m) != v)
cout<<-1<<'\n';
else cout<<l<<'\n';
}
}
return 0;
}