Cod sursa(job #909177)

Utilizator TeodoraTanaseTeodora Tanase TeodoraTanase Data 10 martie 2013 14:40:43
Problema Arbori indexati binar Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.27 kb
#include <cstdio>

#define NMAX 100001

using namespace std;

int n;
int aib[NMAX];

int zeros(int x)
{
    int k = 0;

    for(; ((1 << k) <= x) && (((1 << k) & x) == 0); ++ k);

    return k;
}

void add(int poz, int val)
{
    aib[poz] += val;

    int k = (1 << zeros(poz));

    if(poz + k <= n)
        add(poz + k, val);
}

void readArray()
{
    int x;

    for(int i = 1; i <= n; ++ i)
    {
        scanf("%d ", &x);
        add(i, x);
    }
}

int sum(int poz)
{
    int k = (1 << zeros(poz));

    if(poz - k > 0)
        return aib[poz] + sum(poz - k);

    return aib[poz];
}

int find_(int a)
{
    for(int k = 1; k <= n; ++ k)
        if(sum(k) == a)
            return k;

    return -1;
}

int main()
{
    int m;
    int c;
    int a;
    int b;

    freopen("aib.out", "w", stdout);
    freopen("aib.in", "r", stdin);
    scanf("%d %d\n", &n, &m);

    readArray();

    while(m --)
    {
        scanf("%d %d ", &c, &a);

        if(c != 2)
        {
            scanf("%d ", &b);

            if(c == 0)
                add(a, b);
            else
                printf("%d\n", sum(b) - sum(a - 1));
        }
        else
            printf("%d\n", find_(a));
    }

    return 0;
}