Cod sursa(job #2289293)

Utilizator tudor.bujdeiTudor Bujdei-Leonte tudor.bujdei Data 24 noiembrie 2018 12:44:54
Problema Arbori indexati binar Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.19 kb
#include <bits/stdc++.h>
#define NMAX 100005

using namespace std;

ifstream ifs("aib.in");
ofstream ofs("aib.out");

int a[NMAX], n, m;

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

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

int caut_bin(int k)
{
    int l=1, r=n, mid, val;
    while(l<=r)
    {
        mid=(l+r)/2;
        val=query(mid);
        if(val==k) return mid;
        if(val<k) l=mid+1;
        else r=mid-1;
    }
    return -1;
}

int main()
{

    ifs>>n>>m;
    int x, y, op, k;
    for(int i=1;i<=n;i++)
    {
        int x;
        ifs>>x;
        update(i, x);
    }

    for(int i=1;i<=m;i++)
    {
        ifs>>op;
        if(op==0)
        {
            ifs>>x>>y;
            update(x, y);
        }
        else
            if(op ==1)
        {
            ifs>>x>>y;
            ofs<<query(y)-query(x-1)<<'\n';
        }
        else
            if(op==2)
        {
            ifs>>k;
            ofs<<caut_bin(k)<<'\n';
        }
    }

    return 0;
}