Pagini recente » Cod sursa (job #1532592) | Cod sursa (job #1873148) | Cod sursa (job #1158749) | Cod sursa (job #1553639) | Cod sursa (job #2132085)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
class AIB
{
public:
AIB(int sz)
{
v.resize(sz + 1);
n = sz;
for(int i = 1; i <= n; ++i)
v[i] = 0;
}
void Add(int ind, int val)
{
while(ind <= n)
{
v[ind] += val;
ind += ind & (-ind);
}
}
int Query(int dr)
{
int ret = 0;
while(dr >= 1)
{
ret += v[dr];
dr -= dr & (-dr);
}
return ret;
}
int Query(int st, int dr)
{
return Query(dr) - Query(st - 1);
}
int GetFirstPos(int sum)
{
int step;
for(step = 1; step < n; step <<= 1);
for(int i = 0; step > 0; step >>= 1)
{
if(i + step <= n)
{
if(sum >= v[i + step])
{
i += step;
sum -= v[i];
if(sum == 0)
return i;
}
}
}
return -1;
}
private:
vector<int> v;
};
const int nMax = 100005;
int n, m;
int main()
{
ifstream in("aib.in");
ofstream out("aib.out");
in >> n >> m;
AIB aib(n);
int x;
for(int i = 1; i <= n; ++i)
{
in >> x;
aib.Add(i, x);
}
int t, a, b;
for(int i = 1; i <= m; ++i)
{
in >> t;
if(t == 0)
{
in >> a >> b;
aib.Add(a, b);
}
if(t == 1)
{
in >> a >> b;
out << aib.Query(a, b) << "\n";
}
if(t == 2)
{
in >> a;
out << aib.GetFirstPos(a) << "\n";
}
}
return 0;
}