Cod sursa(job #2521275)

Utilizator AndreiDeltaBalanici Andrei Daniel AndreiDelta Data 10 ianuarie 2020 18:14:21
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <bits/stdc++.h>
#define Dim 15001
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
long long aib[Dim],N,M,A[Dim],S[Dim],op,a,b;

int Calc(int x)
{
    int suma=0;
    while(x)
    {
        suma+=aib[x];
        x-= x & -x;
    }
    return suma;
}

void Update()
{
    int x=a;
    while(x<=N)
    {
        aib[x]-=b;
        x+= x & -x;
    }
}

int main()
{
    f>>N>>M;
    for(int i=1;i<=N;i++)
    {
        f>>A[i];
        S[i]=S[i-1]+A[i];
        int p2= i & -i;
        aib[i]=S[i]-S[ i - p2 ];
    }

    for(int i=1;i<=M;i++)
    {
        f>>op>>a>>b;
        if(op==1)  g<<Calc(b)-Calc(a-1)<<'\n';
        else Update();

    }

    return 0;
}