Cod sursa(job #2592316)

Utilizator rares9991Matisan Rares-Stefan rares9991 Data 1 aprilie 2020 16:08:52
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.23 kb
#include <bits/stdc++.h>

using namespace std;

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

const int N=1 << 16;

int t[N];
int poz, val;

void actualizare(int p, int st, int dr)
{
    if(st==dr)
    {
        if(t[p]==0)
            t[p]=val;
        else
        t[p]-=val;
        return;
    }
    int m=(st+dr)/2;
    if(poz<=m)
    {
        actualizare(2*p, st, m);
    }
    else
        actualizare(2*p+1, m+1, dr);
    t[p]=t[2*p]+t[2*p+1];
}

int a, b;

int interogare(int p, int st, int dr)
{
    if(a<=st && dr<=b)
        return t[p];
    int m=(st+dr)/2, mst=0, mdr=0;
    if(b>m)
        mdr=interogare(2*p+1, m+1, dr);
    if(a<=m)
        mst=interogare(2*p, st, m);
    return mst+mdr;
}

int main()
{
    int n, q;
    in>>n>>q;
    for(int i=1; i<=n; i++)
    {
        in>>val;
        poz=i;
        actualizare(1,1,n);
    }
    for(int i=1; i<=q; i++)
    {
        int c;
        in>>c;
        if(c==1)
        {
            in>>a>>b;
            if(a>b)
                swap(a,b);
            out<<interogare(1,1,n)<<"\n";
        }
        else
        {
            in>>poz>>val;
            actualizare(1,1,n);
        }
    }
    return 0;
}