Cod sursa(job #2450614)

Utilizator stefanpiturStefan Alexandru Pitur stefanpitur Data 23 august 2019 19:28:48
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <iostream>
#include <cstdio>
using namespace std;
const int NMAX = 15001;
int aib[NMAX],aibSize;
void add(int poz, int val){
    do{
        aib[poz] += val;
        poz += poz & (-poz);
    }while(poz <= aibSize);
}
int sum(int poz){
    int result = 0;
    while(poz){
        result += aib[poz];
        poz -= poz & (-poz);
    }
    return result;
}
int main()
{
    FILE *fin, *fout;
    int n,m,i,op,a,b;
    fin = fopen("datorii.in","r");
    fout = fopen("datorii.out","w");
    fscanf(fin,"%d %d",&n,&m);
    aibSize = n;
    for(i=1;i<=n;i++){
        fscanf(fin,"%d",&a);
        add(i,a);
    }
    for(i=1;i<=m;i++){
        fscanf(fin,"%d %d %d",&op,&a,&b);
        if(op == 1)
            fprintf(fout,"%d\n",sum(b) - sum(a - 1));
        else
            add(a,-b);
    }
    fclose(fin);
    fclose(fout);
    return 0;
}