Cod sursa(job #2426094)

Utilizator mihaimodiMihai Modi mihaimodi Data 26 mai 2019 11:00:49
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int a[100001],aib[100001];
int op,m,n,x,y;
int query(int x)
{
    int s=0;
    while(x>0)
    {
        s+=aib[x];
        x-=(x&(-x));
    }
    return s;
}
int cautbin(int val)
{
    int st=1,dr=n,x;
    while(st<=dr)
    {
        int mij=(st+dr)/2;
        int v=query(mij);
        if(val==v)
            return mij;
        else if(val>v)
            st=mij+1;
        else
            dr=mij-1;

    }
    return -1;
}
void update(int x,int val)
{
    while(x<=n)
    {
        aib[x]+=val;
        x+=(x&(-x));
    }
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>a[i];
        update(i,a[i]);
    }

    for(int i=1;i<=m;i++)
    {
        fin>>op>>x;
        if(op==0)
        {
            fin>>y;
            update(x,y);
        }
        else if(op==1)
        {
            fin>>y;
            fout<<query(y)-query(x-1)<<'\n';
        }
        else
        {
            fout<<cautbin(x)<<'\n';
        }
    }
    return 0;
}