Cod sursa(job #2827366)

Utilizator fanevodaCalota Stefan fanevoda Data 5 ianuarie 2022 17:35:18
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1 kb
#include <iostream>
#include <fstream>
using namespace std;

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

int aib[15005];
int n;
int suma(int x)
{
    int s=0;
    for (int i= x;i>0;i-=(i&-i))
        s+= aib[i];
    return s;
}

int sumaintervale(int x, int y)
{
    return suma(y)-suma(x-1);
}

void adaugavaloare(int x,int val)
{
   for (int i=x;i<=n;i+=(i&-i))
       aib[i]+=val;
}

int main()
{
    long long m;
    in >>n;
    in >>m;

    for (int i=1;i<=n;i++)
    {
        aib[i]=0;
    }
    int x;
    for (int i=1;i<=n;i++)
    {
        in>>x;
        adaugavaloare(i,x);
    }

    for (int i=1;i<=m;i++)
    {
        int A;
        in >>A;
        if (A==1)
        {
            int st,dr;
            in >>st>>dr;
            out << sumaintervale(st,dr)<<endl;
        }
        if (A==0)
        {
            int zi,valoare;
            in >> zi >> valoare;
            adaugavaloare(zi,-valoare);
        }
    }
    return 0;
}