Cod sursa(job #3326510)

Utilizator Victor321321Victor Casandra Victor321321 Data 29 noiembrie 2025 11:42:08
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.27 kb
#include <bits/stdc++.h>

using namespace std;

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

struct SegTree
{
    int n;
    vector<long long> tree;
    void init(int _n)
    {
        n=_n;
        tree.assign(4 * n + 4, 0);
    }
    void build(const vector<long long> &a, int node, int l, int r)
    {
        if (l == r)
        {
            tree[node] = a[l];
            return;
        }
        int mid = (l + r) / 2;
        build(a, node * 2, l, mid);
        build(a, node * 2 + 1, mid + 1, r);
        tree[node] = tree[node * 2] + tree[node * 2 + 1];
    }
    void build(const vector<long long> &a)
    {
        build(a, 1, 1, n);
    }
    void pointUpdate(int node, int l, int r, int pos, long long val)
    {
        if (l == r)
        {
            tree[node] -= val;
            return;
        }
        int mid = (l + r) / 2;
        if (pos <= mid)
            pointUpdate(node * 2, l, mid, pos, val);
        else
            pointUpdate(node * 2 + 1, mid + 1, r, pos, val);
        tree[node] = tree[node * 2] + tree[node * 2 + 1];
    }
    void pointUpdate(int pos, long long val)
    {
        pointUpdate(1, 1, n, pos, val);
    }
    long long rangeQuery(int node, int l, int r, int ql, int qr)
    {
        if (qr < l || r < ql)
        {
            return 0;
        }
        if (ql <= l && r <= qr)
        {
            return tree[node];
        }
        int mid = (l + r) / 2;
        return rangeQuery(node * 2, l, mid, ql, qr) +
               rangeQuery(node * 2 + 1, mid + 1, r, ql, qr);
    }
    long long rangeQuery(int l, int r)
    {
        return rangeQuery(1, 1, n, l, r);
    }
};
int main()
{
    int n, q;
    fin >> n >> q;
    vector<long long> a(n + 1);
    for (int i = 1; i <= n; i++)
    {
        fin >> a[i];
    }
    SegTree ST;
    ST.init(n);
    ST.build(a);
    while (q--)
    {
        int type;
        fin >> type;
        if (type == 0)
        {
            int pos;
            long long x;
            fin >> pos >> x;
            ST.pointUpdate(pos, x);
        }
        else if (type == 1)
        {
            int l, r;
            fin >> l >> r;
            fout << ST.rangeQuery(l, r) << "\n";
        }
    }
    return 0;
}