Cod sursa(job #2221445)

Utilizator MocalinnoMoca Andrei Catalin Mocalinno Data 14 iulie 2018 12:55:05
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.36 kb
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
#define zeros(x) ((x ^ (x - 1)) & x)
int n, m, cer, x, a, b, st, dr, mij, poz, aib[100002];
void update(int poz, int val)
{
    for (int i = poz; i <= n; i += zeros(i))
        aib[i] += val;
}
int sum(int x)
{
    int s = 0;
    for (int i = x; i; i -= zeros(i))
        s += aib[i];
    return s;
}
int main()
{
    fin >> n >> m;
    for (int i = 1; i <= n; ++i)
    {
        fin >> x;
        update(i, x);
    }
    for (int i = 1; i <= m; ++i)
    {
        fin >> cer >> a;
        if (cer == 0)
        {
            fin >> b;
            update(a, -b);
        }
        else if (cer == 1)
        {
            fin >> b;
            fout << sum(b) - sum(a - 1) << '\n';
        }
        else
        {
            st = 1;
            dr = n;
            poz = -1;
            while (st <= dr)
            {
                mij = (st + dr) / 2;
                if (sum(mij) < a)
                    st = mij + 1;
                else if (sum(mij) > a)
                    dr = mij - 1;
                else
                {
                    poz = mij;
                    dr = mij - 1;
                }
            }
            fout << poz << '\n';
        }
    }
    fin.close();
    fout.close();
    return 0;
}