Cod sursa(job #1218761)

Utilizator gabrielinelusGabriel-Robert Inelus gabrielinelus Data 12 august 2014 14:40:30
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2 kb
#include <cstdio>

#define Nmax 100005
#define nextp ((p^(p-1))&p)

using namespace std;

#define DIM 10666
int poz = DIM - 1;
char buff[DIM];

void scan(int &A)
{
    A = 0;
    while('0' > buff[poz] || buff[poz] >'9')
        if(++poz == DIM) fread(buff,1,DIM,stdin),poz=0;
    while('0' <= buff[poz] && buff[poz] <= '9')
    {
        A = A * 10 + buff[poz] - 48;
        if(++poz == DIM) fread(buff,1,DIM,stdin),poz=0;
    }
}

int N,M,a,b,op;
class FenwickTree{
private:
    int range[Nmax];
    int answer;
public:
    void Update(int pz,int val)
    {
        for(int p = pz; p <= N; p += nextp)
            range[p] += val;
    }
    int Querry(int pz)
    {
        int answer = 0;
        for(int p = pz; p; p -= nextp)
            answer += range[p];
        return answer;
    }
}Aib;

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);
        Aib.Update(i,a);
    }
    for(int i = 1; i <= M; ++i)
    {
        scanf("%d",&op);
        if(op == 0)
        {
            scanf("%d%d",&a,&b);
            Aib.Update(a,b);
        }
        else
            if(op == 1)
            {
                scanf("%d%d",&a,&b);
                printf("%d\n",Aib.Querry(b) - Aib.Querry(a-1));
            }
            else
            {
                int li = 1,lf = N,m,done = 0;
                scanf("%d",&a);
                while(li <= lf)
                {
                    m = li + ((lf - li) >> 1);
                    int x = Aib.Querry(m);
                    if(x == a){
                        printf("%d\n",m);
                        done = 1;
                    }
                    if(x > a)
                        lf = m - 1;
                    else
                        li = m + 1;
                }
                if(done) continue;
                printf("-1\n");
            }
    }
    return 0;
}