Cod sursa(job #1770106)

Utilizator PletoPletosu Cosmin-Andrei Pleto Data 3 octombrie 2016 19:22:01
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include <fstream>

using namespace std;

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

int tree[100001];
int N, M, op;

void update(int idx, int val)
{
    while(idx<=N)
    {
        tree[idx] += val;
        idx += (idx & -idx);
    }
}

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

int search(int Sum)
{
    int poz=N+1, B=N;
    int st=0, dr=N+1;

    if(query(B) == Sum) poz=N;

    while(B)
    {
          B=(st+dr)>>1;

          if(query(B) > Sum )
          {
               if (dr > B) dr=B;
               B -= 1;
          }
          else if(query(B) == Sum )
          {
              poz=min(poz ,B);
              dr=B;
              B -= 1;
          }
          else
          {
              if(st < B) st=B;
              B += 1;
          }

          if(B <= st) break;
          if(B >= dr) break;
    }

    if (poz == N+1) return -1;
    return poz;
}

int main()
{
    fin>>N>>M;
    for(int i=1; i<=N; ++i)
    {
        int x;
        fin>>x;
        update(i, x);
    }
    for(int i=1; i<=M; ++i)
    {
        int x, a, b;
        fin>>x;
        switch(x)
        {
        case 0:
            fin>>a>>b;
            update(a, b);
            break;
        case 1:
            fin>>a>>b;
            fout<<query(b) - query(a-1)<<'\n';
            break;
        case 2:
            fin>>a;
            fout<<search(a)<<'\n';
            break;
        };
    }
    return 0;
}