Cod sursa(job #2753775)

Utilizator Virgil993Virgil Turcu Virgil993 Data 24 mai 2021 13:22:21
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <iostream>
#include<bits/stdc++.h>

using namespace std;

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

#define zeros(x)((x ^(x - 1))& x )

//Folosim arbori indexati binar pentru a rezolva a doua parte a problemei in complexitate O(logn)
//S este sirul din care memoram variabilele de pe al doilea rand din fisier
//Vom folosi vectorul AIB(x) pentru a memora suma subsecventei din sirul S cu indicii [x-2^k+1,x] unde k = numarul de 0-uri terminale din reprezentarea binara a lui x
//Zeros este o functie ce ne ajuta sa gasim 2^k foarte repede. Astfel atunci cand trebuie sa decrementam un numar din S de pe pozitia j , putem sa decrementam doar sumele subscventelor
//ce il contin pe j.Functia Add incrementeaza/decrementeaza un numar de pe un anumit indice, functia compute returneaza suma subsecventei de la 1,la indicele ales

int AIB[20000];
int v[20000];

void Add(int x, int quantity,int n)
{

    int i;
    for (i = x; i <= n; i += zeros(i))
        AIB[i] += quantity;

}
int Compute(int x)
{
    int i, ret = 0;
    for (i = x; i > 0; i -= zeros(i))
        ret += AIB[i];
    return ret;

}

int main()
{
    int n,m;
    int com;
    int x,y;
    fin>>n;
    fin>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>v[i];
        Add(i,v[i],n);
    }
    for(int i=0;i<m;i++)
    {
        fin>>com;
        if(com == 1)
        {
            fin>>x>>y;
            fout<<Compute(y)-Compute(x-1)<<"\n";
        }
        else
        {
            fin>>x>>y;
            Add(x,-y,n);
        }
    }



    return 0;
}