Cod sursa(job #2527394)

Utilizator RedXtreme45Catalin RedXtreme45 Data 20 ianuarie 2020 11:30:03
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <fstream>
#define Nmax 15001
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int n,m,v[Nmax],aib[Nmax];
void creare(int poz,int val)
{
    while (poz<=n)
    {
        aib[poz]+=val;
        poz+=(poz&(-poz));
    }
}
int suma(int poz){
    int S=0;
    while (poz>0){
        S+=aib[poz];
        poz-=(poz&(-poz));
    }
    return S;
}
int main()
{
    int p,a,b,i,j;
    fin>>n>>m;
    for (i=1;i<=n;i++)
    {
        fin>>v[i];
        creare(i,v[i]);
    }
    for (i=1;i<=m;i++)
    {
        fin>>p;
        fin>>a>>b;
        if (p==1)
        {
            a--;
            fout<<suma(b)-suma(a)<<"\n";
        }
        if (p==0)
        {
            v[a]-=b;
            b=-b;
            creare(a,b);
        }
    }
    return 0;
}