Cod sursa(job #3254401)

Utilizator Stormtrooper-007Vartic Rihard Stormtrooper-007 Data 7 noiembrie 2024 15:28:16
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <fstream>
using namespace std;

ifstream in ("datorii.in");
ofstream out ("datorii.out");

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

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

int rangeSum(int a, int b)
{
    return ps(b)-ps(a-1);
}

int main()
{
    in>>n>>m;
    for (int i=1; i<=n; i++)
    {
        int a;
        in>>a;
        update(i,a);
    }
    for (int i=0; i<m; i++)
    {
        int x,y;
        int c;
        in>>c>>x>>y;
        if (c==0)
        {
            update(x,-y);
            continue;
        }
        out<<rangeSum(x,y)<<endl;
    }
    return 0;
}