Pagini recente » Cod sursa (job #1818447) | Cod sursa (job #2267261) | Cod sursa (job #2267638) | Cod sursa (job #1433944) | Cod sursa (job #2740918)
#include <bits/stdc++.h>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n, m;
int v[100001], arb[100001];
void init()
{
int i;
for(i=1; i<=n; i++)
arb[i]=v[i];
for(i=1; i<=n; i++)
{
int j=i+(i&-i);
arb[j]+=arb[i];
}
}
int sum(int i)
{
int s=0;
while(i)
{
s+=arb[i];
i-=(i&-i);
}
return s;
}
int query(int st, int dr)
{
return sum(dr)-sum(st-1);
}
void update(int i, int k)
{
while(i<=n)
{
arb[i]+=k;
i+=(i&-i);
}
}
int query1(int k)
{
int p=-1;
int st=1, dr=n;
while(st<=dr)
{
int mij=(st+dr)/2;
int s=sum(mij);
if(s==k)
{
p=mij;
dr=mij-1;
}
else if(s>k)
dr=mij-1;
else
st=mij+1;
}
return p;
}
int main()
{
int i, op, a, b;
fin >> n >> m;
for(i=1; i<=n; i++)
fin >> v[i];
init();
for(i=1; i<=m; i++)
{
fin >> op;
if(op==0)
{
fin >> a >> b;
update(a, b);
}
else if(op==1)
{
fin >> a >> b;
fout << query(a, b) << '\n';
}
else
{
fin >> a;
fout << query1(a) << '\n';
}
}
return 0;
}