Cod sursa(job #2509679)

Utilizator Danut200333Dumitru Daniel Danut200333 Data 14 decembrie 2019 15:37:37
Problema Arbori indexati binar Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include <fstream>

using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int aib[100001],i,n,m,c,a,b,x,st,dr,mid;
void update(int a,int b)
{
    int i;
    for(i=a;i<=n;i+=(i&(-i)))aib[i]+=b;
}
int suma(int x)
{
    int i,s=0;
    for(i=x;i>0;i-=(i&(-i)))s+=aib[i];
    return s;
}

int main()
{
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>x;
        update(i,x);
    }
    for(i=1;i<=m;i++)
    {
        fin>>c;
        if(c==0||c==1)fin>>a>>b;
        else fin>>a;
        if(c==0)update(a,b);
        if(c==1)fout<<suma(b)-suma(a-1)<<'\n';
        if(c==2){st=0;dr=n;while(st<=dr)
            {
                mid=(st+dr)/2;
                if(suma(mid)==a){break;}
                if(suma(mid)>a)dr=mid-1;
                if(suma(mid)<a)st=mid+1;

            }
            if(suma(mid)!=a)fout<<-1<<'\n';
            else    fout<<mid<<'\n';
        }
    }
    return 0;
}