Cod sursa(job #1826241)

Utilizator andysoloAndrei Solo andysolo Data 10 decembrie 2016 11:43:40
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.27 kb
#include<bits/stdc++.h>
# define ub(x) (x&(-x))
using namespace std;

int N,M,x;
long long int AiB[100005];
int d,p,k;

int lbit(int x)
{
    return x&(x-1);
}

int binc(int s,int k)
{
    if(k%2==1 || (k==N && s>AiB[k]))
        return -1;
    else if(s>AiB[k])
        return binc(s-AiB[k],k+lbit(k)/2);
    else if(s<AiB[k])
        return binc(s,k-lbit(k)/2);
    else if(s==AiB[k])
        return k;
}

void addAiB(int x,int p)
{
    for(int i=p; i<=N; i+=lbit(i))
        AiB[i]+=x;
}

long long int getAiB(int p)
{
    long long int s=0;
    for(int i=p; i>0; i-=lbit(i))
        s+=AiB[i];
    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",&x);
        addAiB(x,i);
    }

    for(int i=1; i<=M; i++)
    {
        scanf("%d",&d);
        if(d==0)
        {
            scanf("%d %d",&p,&k);
            addAiB(k,p);
        }
        else if(d==1)
        {
            scanf("%d %d",&p,&k);
            printf("%llid \n",getAiB(k)-getAiB(p-1));
        }
        else if(d==2)
        {
            scanf("%d",&p);
     //       printf("%d \n",binc(p,N));
        }
    }

    return 0;
}