Cod sursa(job #1605367)

Utilizator PaulCbnCiobanu Paul PaulCbn Data 18 februarie 2016 22:40:04
Problema Arbori indexati binar Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include <iostream>
#include <cstdio>
using namespace std;

int v[100500],n,m,a,b,type;

void actualizare(int index, int valoare)
{
    while(index<=n)
    {
        v[index]+=valoare;
        int p = ((index^(index - 1))+1)>>1;
        index += p;
    }
}

int suma(int index)
{
    int s=0;
    while(index)
    {
        s+=v[index];
        index = index&(index-1);
    }
    return s;
}

int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);

    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a);
        actualizare(i,a);
    }

    for(int i=1;i<=m;i++)
    {
        scanf("%d",&type);
        if(type == 0)
        {
            scanf("%d%d",&a,&b);
            actualizare(a,b);

        }
        else if(type == 1)
        {
            scanf("%d%d",&a,&b);
            printf("%d\n",suma(b)-suma(a-1));
        }
        else
        {
            int j;
            scanf("%d",&a);
            for(j=1;j<=n;j++)
                if(suma(j)==a)
                {
                    printf("%d\n",j);
                    break;
                }
            if(j>n)
                printf("-1\n");
        }

    }



    return 0;
}