Cod sursa(job #1970561)

Utilizator VladG26Ene Vlad-Mihai VladG26 Data 19 aprilie 2017 14:06:24
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.46 kb
#include <iostream>
#include <cstdio>
#define nr_zero(x) x&(x^(x-1))
using namespace std;
int n,m,AIB[100003];
void update_arb(int POZ,int VAL)
{
    for(int IT=POZ;IT<=n;IT+=nr_zero(IT))
    {
        AIB[IT]+=VAL;
    }
}
int querySUM_arb(int INDEX)
{
    int rez=0;

    for(int IT=INDEX;IT>=1;IT-=nr_zero(IT))
        rez+=AIB[IT];

    return rez;
}
void query1_arb(int ST,int DR)
{
    int rez;
    rez=querySUM_arb(DR)-querySUM_arb(ST-1);
    printf("%d\n",rez);
}
void query2_arb(int VAL)
{
    int st=1,dr=n,aux,mij;
    while(st<=dr)
    {
        mij=(st+dr)/2;
        aux=querySUM_arb(mij);
        if(aux==VAL)
        {
            printf("%d\n",mij);
            return;
        }
        if(aux<VAL)
            st=mij+1;
        else
            dr=mij-1;
    }
    printf("-1\n");
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    int aux;
    int p1,p2,p3;
    scanf("%d%d",&n,&m);

    for(int i=1;i<=n;i++)
    {
        scanf("%d",&aux);
        update_arb(i,aux);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&p1);
        if(p1==0)
        {
            scanf("%d%d",&p2,&p3);
            update_arb(p2,p3);
        }
        if(p1==1)
        {
            scanf("%d%d",&p2,&p3);
            query1_arb(p2,p3);
        }
        if(p1==2)
        {
            scanf("%d",&p2);
            query2_arb(p2);
        }
    }
    return 0;
}