Cod sursa(job #2349945)

Utilizator radurotaruRotaru Radu Stefan radurotaru Data 20 februarie 2019 21:14:48
Problema Arbori indexati binar Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
#include <fstream>

using namespace std;

ifstream f("aib.in");
ofstream g("aib.out");

int BITree[100005],n,x,m,a,b;

int getSum(int q)
{
    int s=0;
    while(q)
    {
        s=s+BITree[q];
        q-=q &(-q);
    }
    return s;
}
void update(int q, int val)
{
    while(q<= n)
    {
        BITree[q]+=val;
        q+=q &(-q);
    }
}
int Search()
{
    for(int i=1; i<=n; i++)
    {
        int s=0;
        int q=i;
        while(q)
        {
            s=s+BITree[q];
            q-=q &(-q);
            if(s>a)
                break;
        }
        if(s==a)
            return i;
    }
    return -1;
}
int main()
{
    f>>n>>m;
    for(int i=1; i<=n; i++)
    {
        f>>x;
        update(i, x);
    }

    for(int i=1; i<=m; i++)
    {
        f>>x;
        if(x==0)
        {
            f>>a>>b;
            update(a, b);
        }
        if(x==1)
        {
            f>>a>>b;
            g<<getSum(b)-getSum(a-1)<<'\n';
        }
        if(x==2)
        {
            f>>a;
            g<<Search()<<'\n';
        }
    }
    return 0;
}