Cod sursa(job #3138399)

Utilizator AnonymousUserBogdan Ionelia AnonymousUser Data 19 iunie 2023 14:12:33
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.74 kb
#include <iostream>
#include <fstream>

using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
const int N = 150001;
int arr[N];
int segment_tree[4*N];


void build(int node, int left, int right)
{
    if(left == right)
        segment_tree[node] = arr[left];
    else
    {
        int m = (left + right) / 2;
        build(node*2, left, m);
        build(node*2+1, m+1, right);
        segment_tree[node] = segment_tree[node*2] + segment_tree[node*2+1];
    }
}

void update(int node, int left, int right, int pos, int new_value)
{
    if(left == right)
        segment_tree[node] -= new_value;
    else
    {
        int m = (left + right) / 2;
        if(pos <= m)
            update(node*2, left, m, pos, new_value);
        else
            update(node*2+1, m+1, right, pos, new_value);
        segment_tree[node] = segment_tree[node*2] + segment_tree[node*2+1];
    }
}

int query(int node, int left, int right, int q_left, int q_right)
{
    if(q_left <= left && right <= q_right)
        return segment_tree[node];
    else
    {
        int m = (left + right) / 2;
        if(q_right <= m)
            return query(node*2, left, m, q_left, q_right);
        if(m+1 <= q_left)
            return query(node*2+1, m+1, right, q_left, q_right);
        return query(node*2, left, m, q_left, q_right) + query(node*2+1, m+1, right, q_left, q_right);
    }
}

int main()
{
    int n, m, task, a, b;
    fin >> n >> m;
    for(int i = 1; i <= n; ++i)
        fin >> arr[i];
    build(1, 1, n);
    while(m--)
    {
        fin >> task >> a >> b;
        if(task)
            fout << query(1, 1, n, a, b) << '\n';
        else
            update(1, 1, n, a, b);
    }
    return 0;
}