Cod sursa(job #1828479)

Utilizator andysoloAndrei Solo andysolo Data 13 decembrie 2016 13:52:16
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 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 binc(int s,int k)
{
    if(s==AiB[k])
        return k;
    else if(k%2==1 || (k==N && s>AiB[k]))
        return -1;
    else if(s>AiB[k])
        return binc(s-AiB[k],k+ub(k)/2);
    else if(s<AiB[k])
        return binc(s,k-ub(k)/2);
}

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

long long int getAiB(int p)
{
    long long int s=0;
    for(int i=p; i>0; i-=ub(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("%lli \n",getAiB(k)-getAiB(p-1));
        }
        else if(d==2)
        {
            scanf("%d",&p);
            printf("%d \n",binc(p,N));
        }
    }

    return 0;
}