Cod sursa(job #2031289)

Utilizator KOzarmOvidiu Badea KOzarm Data 2 octombrie 2017 23:01:19
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.4 kb
#include <fstream>

using namespace std;

ifstream fin("aib.in");
ofstream fout("aib.out");

int a[100001],n,m,x,y,p,sol;

void adauga(int poz,int val)
{
    while(poz<=n)
    {
        a[poz]+=val;
        poz=poz+(poz&(-poz));
    }
}

int suma(int poz)
{
    int sum=0;
    while(poz>0)
    {
        sum+=a[poz];
        poz=(poz&(poz-1));
    }
    return sum;
}

void pozitie(int sum,int &sol)
{
    int poz=1;
    while(poz<=n)
        poz=(poz<<1);
    int in=1,fi=poz;
    while(in!=fi)
    {
        int poz1=(in+fi)/2;
        if(poz1>n)
            fi=poz1;
        else
        if(a[poz1]>sum)
            fi=poz1;
        else
        if(a[poz1]==sum)
        {
            sol=poz1;
            return;
        }
        else
        {
            sum-=a[poz1];
            in=poz1+1;
        }
    }
}



int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>x;
        adauga(i,x);
    }
    while(m)
    {
        m--;
        fin>>p;
        if(p==0)
        {
            fin>>x>>y;
            adauga(x,y);
        }
        else
        if(p==1)
        {
            fin>>x>>y;
            fout<<suma(y)-suma(x-1)<<"\n";
        }
        else
        if(p==2)
        {
            fin>>x;
            sol=-1;
            pozitie(x,sol);
            fout<<sol<<"\n";
        }
    }
    return 0;
}