Cod sursa(job #2921326)

Utilizator AndreiBOTOBotocan Andrei AndreiBOTO Data 30 august 2022 12:06:53
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <iostream>
#include <fstream>
#include <algorithm>

using namespace std;

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

const int NMAX=1e5+5;
int v[NMAX];
int aib[NMAX];
int n;

int lsb(int x)
{
    return x^(x&(x-1));
}

void init(int p,int val)
{
    while(p<=n)
    {
        aib[p]+=val;
        p+=lsb(p);
    }
}

void update(int p,int val)
{
    v[p]=v[p]-val;
    while(p<=n)
    {
        aib[p]-=val;
        p+=lsb(p);
    }
}

long long solve(int p)
{
    long long s=0;
    while(p>0)
    {
        s+=aib[p];
        p-=lsb(p);
    }
    return s;
}

int main()
{
    int m,i,j,t,cod,ve,p,q;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>v[i];
        init(i,v[i]);
    }
    for(i=1;i<=m;i++)
    {
        fin>>cod;
        if(cod==0)
        {
            fin>>t>>ve;
            update(t,ve);
        }
        else
        {
            fin>>p>>q;
            fout<<solve(q)-solve(p-1)<<"\n";
        }
    }
    return 0;
}