Cod sursa(job #1829115)

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

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

long long int max2(long long int x) {
    long long int p=1;
    while(p*2<=x)
        p*=2;
    return p;
}

long long int binc(long long int s,long long int k) {
    if(s==AiB[k])
        return k;
    else if(k%2==1 || k>N)
        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(long long int x,long long int p) {
    for(long long int i=p; i<=N; i+=ub(i))
        AiB[i]+=x;
}

long long int getAiB(long long int p) {
    long long int s=0;
    for(long long 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("%lld %lld",&N,&M);

    for(long long int i=1; i<=N; i++) {
        scanf("%lld",&x);
        addAiB(x,i);
    }

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

    return 0;
}