Cod sursa(job #3270060)

Utilizator Cristian_NegoitaCristian Negoita Cristian_Negoita Data 21 ianuarie 2025 21:27:06
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
#define lsb(i) (i & (-i))
const int NMAX = 15005;
int n, m, a[NMAX], aib[NMAX];

void update(int day, int val)
{
    for(int i = day; i <= n; i += lsb(i))
        aib[i] += val;
}

int sum(int dr)
{
    int ans = 0;
    for(int i = dr; i >= 1; i -= lsb(i))
        ans += aib[i];
    return ans;
}

signed main()
{
    fin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        fin >> a[i];
        update(i, a[i]);
    }

    while(m--)
    {
        bool op;
        fin >> op;
        if(op == 0)
        {
            int day, val;
            fin >> day >> val;
            update(day, -val);
        }
        else
        {
            int st, dr;
            fin >> st >> dr;
            fout << sum(dr) - sum(st - 1) << "\n";
        }
    }

    return 0;
}