Cod sursa(job #1425305)

Utilizator tac1234Tran Bach Nguyen tac1234 Data 27 aprilie 2015 11:43:48
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <cstdio>
using namespace std;
int n,aib[100005];
inline void update(int poz,int val)
{
    for (; poz<=n; poz+=(poz&-poz))
        aib[poz]+=val;
}
inline int query(int poz)
{
    int s=0;
    for(; poz; poz-=(poz&-poz))
        s+=aib[poz];
    return s;
}
inline int find(int val)
{
    int i,poz;
    for(poz=1; poz<n; poz<<=1);
    for(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%d",&a,&b);
            if (!op)
                update(a,b);
            else
                printf("%d\n",query(b)-query(a-1));
        }
        else
        {
            scanf("%d\n",&a);
            printf("%d\n",find(a));
        }
    }
    return 0;
}