Cod sursa(job #1426037)

Utilizator ialexia_ioanaAlexia Ichim ialexia_ioana Data 28 aprilie 2015 20:43:10
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
#include <cstdio>

int n;

int aib[100001];

int numar(int x)
{
    int i, poz;
    for(poz=1; poz<n; poz<<=1);
    for(i=0; poz; poz/=2)
    {
        if(i+poz<=n)
        {
            if(x>=aib[i+poz])
            {
                i+=poz;
                x-=aib[i];
                if(x==0)
                    return i;
            }
        }
    }
    return -1;
}

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);
            printf("%d\n", numar(a));
        }
    }
    return 0;
}