Cod sursa(job #3252305)

Utilizator adelina_15InfoAdelina Radoi adelina_15Info Data 29 octombrie 2024 10:16:47
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, m;
int rad;
vector<int> v, b, num;

void Scade(int i, int val)
{
    b[i/rad] -= min(val, v[i]);
    v[i] -= min(val, v[i]);
}

int Sum(int x, int y)
{
    int s = 0;
    int i = x;
    while(i % rad != 0 && i <= y)
        s += v[i++];
    while(i < n && i+num[i] <= y)
    {
        s += b[i/rad];
        i += num[i];
    }
    while(i <= y)
        s += v[i++];
    return s;
}

int main()
{
    fin >> n >> m;
    v.resize(n);
    rad = sqrt(n);
    b.resize(rad+1);
    num.resize(rad+1);
    for(int i = 0; i < n; i++)
        fin >> v[i];
    for(int i = 0; i < n; i++)
    {
        b[i/rad] += v[i];
        num[i/rad]++;
    }
    for(int i = 0; i < m; i++)
    {
        int op, x, y;
        fin >> op >> x >> y;
        if(op == 0)
            Scade(x-1, y);
        else
            fout << Sum(x-1, y-1) << "\n";
    }
    return 0;
}