Cod sursa(job #2509318)

Utilizator andreitabaraandrei2004 andreitabara Data 14 decembrie 2019 09:46:20
Problema Arbori indexati binar Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.9 kb
#include <fstream>

using namespace std;
long v[100001],x,i,c,a,b,n,m,j,st,dr,ok,mijl;
ifstream in ("aib.in");
ofstream out ("aib.out");
void update(int x,int p)
{
    int i;
    for(i=p;i<=n;i+=(i&(-i)))
    {
        v[i]+=x;
    }
}

long suma(int p)
{
    int i,s=0;
    for(i=p;i>0;i-=(i&(-i)))
    {
        s+=v[i];
    }

    return s;
}
int main()
{
in>>n>>m;
for(i=1;i<=n;i++)
{
    in>>x;
    update(x,i);
}

for(i=1;i<=m;i++)
{
    in>>c>>a;
    if(c==0){in>>b;update(b,a);}
    else{if(c==1){in>>b;out<<suma(b)-suma(a-1)<<'\n';}
         else{st=0;
         dr=n;
         ok=1;
         while(st<=dr)
         {mijl=(st+dr)/2;
             if(suma(mijl)>a)dr=mijl-1;
             if(suma(mijl)<a)st=mijl+1;
             if(suma(mijl)==a){ok=1;break;}
         }
       if(ok==1) out<<mijl<<'\n';
       else{out<<-1<<'\n';}
         }}

}
    return 0;
}