Cod sursa(job #1425352)

Utilizator ccygnusMaygnus Pop ccygnus Data 27 aprilie 2015 12:54:45
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda pregatire-lot-aib Marime 1.14 kb
#include<cstdio>
using namespace std;
int aib[100005],n;
void update(int poz,int val)
{
    int i;
    for(i=poz;i<=n;i=i+(i&(-i)))
        aib[i]+=val;
}
int query(int k)
{
    int sol=0,s=0,pas;
    for(pas=1<<17;pas>0;pas=pas>>1)
        if(sol+pas<=n && s+aib[sol+pas]<=k)
            {
            sol=sol+pas;
            s=s+aib[sol];
            }
    if (s!=k || sol==0)
        return -1;
    return sol;
}
int query2(int poz)
{
    int i,s=0;
    for(i=poz;i>0;i=i-(i&(-i)))
        s=s+aib[i];
    return s;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    int m,i,x,op,a,b;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
        {
        scanf("%d",&x);
        update(i,x);
        }
    for(i=1;i<=m;i++)
        {
        scanf("%d",&op);
        if (op==0)
            {
            scanf("%d%d",&a,&b);
            update(a,b);
            }
        if (op==1)
            {
            scanf("%d%d",&a,&b);
            printf("%d\n",query2(b)-query2(a-1));
            }
        if (op==2)
            {
            scanf("%d",&a);
            printf("%d\n",query(a));
            }
        }
return 0;
}