Cod sursa(job #1425080)

Utilizator ialexia_ioanaAlexia Ichim ialexia_ioana Data 26 aprilie 2015 17:00:14
Problema Arbori indexati binar Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.22 kb
#include <cstdio>

int n;

int aib[15001];

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;
}

using namespace std;

int main()
{
    freopen("aib.in", "r", stdin);
    freopen("aib.out", "w", stdout);
    int m, a, b, c, tip, i, j, s, ok;
    scanf("%d%d", &n, &m);
    for(i=1; i<=n; i++)
    {
        scanf("%d", &c);
        update(i, c);
    }
    for(j=1; j<=m; j++)
    {
        scanf("%d", &tip);
        if (tip==1)
        {
            scanf("%d%d", &a, &b);
            printf("%d\n", query(b)-query(a-1));
        }
        if (tip==0)
        {
            scanf("%d%d", &a, &b);
            update(a, b);
        }
        if (tip==2)
        {
            scanf("%d", &a);
            ok=1;
            for(i=1; i<=n && ok; i++)
            {
                s=query(i)-query(0);
                if (s==a)
                {
                    printf("%d\n", i);
                    ok=0;
                }
            }
            if (ok)
                printf("-1\n");
        }
    }
    return 0;
}