Pagini recente » Cod sursa (job #2730554) | Cod sursa (job #2879920) | Cod sursa (job #1447844) | Cod sursa (job #1156057) | Cod sursa (job #1770106)
#include <fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int tree[100001];
int N, M, op;
void update(int idx, int val)
{
while(idx<=N)
{
tree[idx] += val;
idx += (idx & -idx);
}
}
int query(int idx)
{
int sum=0;
while(idx>0)
{
sum += tree[idx];
idx -= (idx & -idx);
}
return sum;
}
int search(int Sum)
{
int poz=N+1, B=N;
int st=0, dr=N+1;
if(query(B) == Sum) poz=N;
while(B)
{
B=(st+dr)>>1;
if(query(B) > Sum )
{
if (dr > B) dr=B;
B -= 1;
}
else if(query(B) == Sum )
{
poz=min(poz ,B);
dr=B;
B -= 1;
}
else
{
if(st < B) st=B;
B += 1;
}
if(B <= st) break;
if(B >= dr) break;
}
if (poz == N+1) return -1;
return poz;
}
int main()
{
fin>>N>>M;
for(int i=1; i<=N; ++i)
{
int x;
fin>>x;
update(i, x);
}
for(int i=1; i<=M; ++i)
{
int x, a, b;
fin>>x;
switch(x)
{
case 0:
fin>>a>>b;
update(a, b);
break;
case 1:
fin>>a>>b;
fout<<query(b) - query(a-1)<<'\n';
break;
case 2:
fin>>a;
fout<<search(a)<<'\n';
break;
};
}
return 0;
}