Cod sursa(job #2816555)

Utilizator DajaMihaiDaja Mihai DajaMihai Data 11 decembrie 2021 15:57:46
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <stdio.h>
#include <math.h>

#define MAX_N 15000

const int SQ_N = sqrt(MAX_N);
const int BATOG_SIZE = (MAX_N + SQ_N - 1) / SQ_N;

int v[MAX_N];
int vLength;

int batog[BATOG_SIZE];

void build() {
  int i;

  for (i = 0; i < vLength; ++i)
    batog[i / SQ_N] += v[i];
}

    void update(int pos, int value) {
  batog[pos / SQ_N] -= value;
  v[pos] -= value;
}

int query(int left, int right) {
  int firstInterval, lastInterval, result;

  firstInterval = (left + SQ_N - 1) / SQ_N * SQ_N;
  lastInterval = right / SQ_N * SQ_N;

  result = 0;
  while (left <= right && left < firstInterval)
    result += v[left++];
  while (right >= left && right >= lastInterval)
    result += v[right--];

  firstInterval /= SQ_N;
  lastInterval /= SQ_N;

  while (firstInterval < lastInterval)
    result += batog[firstInterval++];

  return result;
}

int main() {
  FILE *fin, *fout;
  fin = fopen("datorii.in", "r");
  fout = fopen("datorii.out", "w");

  int i, q, op, a, b;

  fscanf(fin, "%d%d", &vLength, &q);
  for (i = 0; i < vLength; ++i)
    fscanf(fin, "%d", &v[i]);

  build();
  while (q--) {
    fscanf(fin, "%d%d%d", &op, &a, &b);

    if (op == 1)
        fprintf(fout, "%d\n", query(a - 1, b - 1));
    else if (op == 0)
        update (a - 1, b);
  }
  return 0;
}