Pagini recente » Cod sursa (job #1939618) | Cod sursa (job #2020299) | Cod sursa (job #1449075) | Cod sursa (job #1937892) | Cod sursa (job #1264141)
#include <fstream>
using namespace std;
ifstream in("aib.in");
ofstream out("aib.out");
const int NMAX = 100000;
int aib[NMAX+2];
int N,Q;
inline int LSB( int nr ) {
return ( nr & (-nr) );
}
void Update_aib( int poz, int val ) {
for( int i= poz; i<=N; i+= LSB(i) ) {
aib[i]+= val;
}
}
int Querry_aib( int poz ) {
int ans= 0;
for( int i= poz; i>0; i-= LSB(i) ) {
ans+= aib[i];
}
return ans;
}
int Bin_search_sum( int val ) {
int step= 1, i= 0, ans= 0;
for( ; step<N; step<<= 1 );
for( ; step; step>>= 1 ) {
if( i+step <= N && ans + aib[step] <= val ) {
i+= step;
ans+= aib[step];
}
}
return i;
}
int main() {
in >> N >> Q;
for( int i= 1; i<=N; ++i ) {
int a;
in >> a;
Update_aib( i, a );
}
for( int i= 1; i<=Q; ++i ) {
int op, x,y,p;
in >> op;
if( op == 0 ) { /// O( log2(N) )
in >> p >> x;
Update_aib( p, x );
}
if( op == 1 ) { /// O( log2(N) )
in >> x >> y;
out << Querry_aib( y ) - Querry_aib( x-1 ) << '\n';
}
if( op == 2 ) { /// O( log2(N) )
in >> x;
out << Bin_search_sum( x ) << '\n';
}
}
return 0;
}