Cod sursa(job #1857118)

Utilizator valentinoMoldovan Rares valentino Data 25 ianuarie 2017 20:29:56
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.89 kb
#include <fstream>
#define zero(x) ((x ^(x-1)) & x)
#define NM 15005
using namespace std;

ifstream f("datorii.in");
ofstream g("datorii.out");
int n, m, aib[NM];

void Update(int val, int poz)
{
    while(poz <= n)
    {
        aib[poz] += val;
        poz += zero(poz);
    }
}

int Sum(int poz)
{
    int s = 0;
    while(poz > 0)
    {
        s += aib[poz];
        poz -= zero(poz);
    }
    return s;
}

void Scade(int val, int poz)
{
    while(poz <= n)
    {
        aib[poz] -= val;
        poz += zero(poz);
    }
}


int main()
{
    int x, query, a, b;
    f >> n >> m;
    for(int i = 1; i <= n; ++i)
    {
        f >> x;
        Update(x, i);
    }
    for(int i = 1; i <= m; ++i)
    {
        f >> query >> a >> b;
        if(query == 0)
        {
            Scade(b, a);
        }
        else g << Sum(b) - Sum(a-1) << '\n';

    }
}