Cod sursa(job #1676092)

Utilizator GinguIonutGinguIonut GinguIonut Data 5 aprilie 2016 18:41:07
Problema Arbori indexati binar Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.51 kb
#include <fstream>
#include <string.h>
#include <vector>
#include <queue>
#include <limits.h>

#define nMax 100001
#define pb push_back
#define INF INT_MAX
#define bit(i) i&(-i)
using namespace std;
ifstream fin("aib.in");
ofstream fout("aib.out");
int n, m;
int aib[nMax], v[nMax];
void upDate(int poz, int val)
{
    for(int i=poz;i<=n;i+=bit(i))
        aib[i]+=val;
}
void read()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>v[i];
        upDate(i, v[i]);
    }
}
int query(int poz)
{
    int Sol=0;
    for(int i=poz;i>0;i-=bit(i))
        Sol+=aib[i];

    return Sol;
}
int verif(int poz)
{
    return query(poz);
}
int binary_search(int val)
{
    int st=0, dr=n, poz;

    while(st<=dr)
    {
        int mid=st+(dr-st)/2;
        if(verif(mid)>=val)
        {
            poz=mid;
            dr=mid-1;
        }
        else
            st=mid+1;
    }

    if(verif(poz)==val)
        return poz;
    return -1;
}
void solve()
{
    int op, a, b;

    for(int i=1;i<=m;i++)
    {
        fin>>op;

        if(op==0)
        {
            fin>>a>>b;
            upDate(a, b);
            continue;
        }

        if(op==1)
        {
            fin>>a>>b;
            fout<<query(b)-query(a-1)<<'\n';
            continue;
        }

        if(op==2)
        {
            fin>>a;
            fout<<binary_search(a)<<'\n';
            continue;
        }
    }
}
int main()
{
    read();
    solve();
    return 0;
}