Cod sursa(job #2119054)

Utilizator ioana_marinescuMarinescu Ioana ioana_marinescu Data 31 ianuarie 2018 16:40:19
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <bits/stdc++.h>
const int MAX_N = 15005;

using namespace std;

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

int n, m, aib[MAX_N];

void add(int poz, int val)
{
    for(int i=poz; i<=n; i+=i&-i)
        aib[i]+=val;
}

int sum(int poz)
{
    int sol=0;
    for(int i=poz; i>=1; i-=i&-i)
        sol+=aib[i];
    return sol;
}
int main()
{
    int i, x, poz, val, a, b, q;
    fin>>n>>m;
    for(i=1; i<=n; i++) {
        fin>>x;
        add(i, x);
    }
    for(i=1; i<=m; i++) {
        fin>>q;
        if(q==0) {
            fin>>poz>>val;
            add(poz, -val);
        }
        if(q==1) {
            fin>>a>>b;
            fout<<sum(b)-sum(a-1)<<'\n';
        }
    }
    return 0;
}