Cod sursa(job #2167526)

Utilizator OlivianOlivian Dan Cretu Olivian Data 13 martie 2018 22:07:52
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include<cstdio>
using namespace std;
int aib[100010],tip,n,m,logn;
void aib_update(int poz,int val)
{
    for(;poz<=n;poz+=poz&(-poz)) aib[poz]+=val;
}
int aib_querry(int poz)
{
    int s=0;
    for(;poz<=n;poz-=poz&(-poz)) s+=poz;
    return poz;
}
int cautbin(int val)
{
    int poz=0;
    for(int i=logn;i>=0;i--)
        if((poz+(1<<i))<=n && aib[poz+(1<<i)]<val)
        {
            poz+=1<<i;
            val-=aib[poz];
        }
    return poz+1;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d %d",&n,&m);
    int logn=0;
    for(;logn<=n;logn<<=1);
    logn--;
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        aib_update(i,x);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&tip);
        int a,b;
        if(tip==0) {scanf("%d %d",&a,&b);aib_update(a,b);}
        else if(tip==1) {scanf("%d %d",&a,&b);printf("%d\n",aib_querry(b)-aib_querry(a-1));}
        else {scanf("%d",&b);a=cautbin(b);if(aib_querry(a)==b) printf("%d\n",a);else printf("-1");}
    }
}