Cod sursa(job #1425344)

Utilizator tac1234Tran Bach Nguyen tac1234 Data 27 aprilie 2015 12:46:11
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda pregatire-lot-aib Marime 1.18 kb
#include <cstdio>
using namespace std;
int n,aib[100005];
inline void update(int poz,int val)
{
    for(register int i=poz; i<=n; i+=(i&-i))
        aib[i]+=val;
}
inline int query(int poz)
{
    int s=0;
    for(register int i=poz; i>=1; i-=(i&-i))
        s+=aib[i];
    return s;
}
inline int find(int val)
{
    int poz;
    for(poz=1; poz<n; poz<<=1);
    for(register int i=0; poz; poz>>=1)
        if (i+poz<=n)
            if(val>=aib[i+poz])
            {
                i+=poz;
                val-=aib[i];
                if (!val)
                    return i;
            }
    return -1;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    int a,b,op,m;
    scanf("%d%d",&n,&m);
    for (register int i=1; i<=n; ++i)
    {
        scanf("%d",&a);
        update(i,a);
    }
    for(register int i=1; i<=m; ++i)
    {
        scanf("%d",&op);
        if (op==2)
        {
            scanf("%d\n",&a);
            printf("%d\n",find(a));
        }
        else
        {
            scanf("%d%d",&a,&b);
            if (op==1)
                printf("%d\n",query(b)-query(a-1));
            else
                update(a,b);
        }
    }
    return 0;
}