Cod sursa(job #1250048)

Utilizator armandpredaPreda Armand armandpreda Data 27 octombrie 2014 19:35:18
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.23 kb
#include <cstdio>

using namespace std;

int n,aib[100005],m;
void update(int poz,int val)
{
    for(;poz<=n;poz+=(poz&(poz-1)^poz))
        aib[poz]+=val;
}
int query(int poz)
{
    int s=0;
    for(;poz;poz-=(poz&(poz-1)^poz))
        s+=aib[poz];
    return s;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    int i;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;++i)
    {
        int x;
        scanf("%d",&x);
        update(i,x);
    }
    for(i=1;i<=m;++i)
    {
        int op,a,b;
        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",query(b)-query(a-1));
        }
        if(op==2)
        {
            scanf("%d",&a);
            int st=1,dr=n,mij,last=-1;
            while(st<=dr)
            {
                mij=(dr-st)/2+st;
                if(query(mij)>=a)
                {
                    dr=mij-1;
                    last=mij;
                }
                else
                    st=mij+1;
            }
            printf("%d\n",last);
        }
    }
    return 0;
}