Cod sursa(job #2636112)

Utilizator BereaBerendea Andrei Berea Data 16 iulie 2020 17:14:10
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.54 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
int n,m,i,t,x,y,sum,mij;
int aint[400004];

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

void update1(int nod, int l, int r, int poz, int val)
{
    if (l==r)
    {
        aint[nod]=val;
        return;
    }
    mij=(l+r)>>1;
    if (poz<=mij) update1(nod*2,l,mij,poz,val);
    else update1(nod*2+1,mij+1,r,poz,val);
    aint[nod]=aint[nod*2]+aint[nod*2+1];
}

void update2(int nod, int l, int r, int poz, int val)
{
    if (l==r)
    {
        aint[nod]-=val;
        return;
    }
    mij=(l+r)>>1;
    if (poz<=mij) update2(nod*2,l,mij,poz,val);
    else update2(nod*2+1,mij+1,r,poz,val);
    aint[nod]=aint[nod*2]+aint[nod*2+1];
}

void query(int nod,int l,int r,int st,int dr)
{
    if (st<=l and r<=dr)
    {
        sum+=aint[nod];
        return;
    }
    int mij=(l+r)>>1;
    if (st<=mij) query(nod*2,l,mij,st,dr);
    if (dr>mij) query(nod*2+1,mij+1,r,st,dr);
}

int main()
{
    cin>>n>>m;
    for (i=1;i<=n;i++)
    {
        cin>>x;
        update1(1,1,n,i,x);
    }
    for (i=1;i<=m;i++)
    {
        cin>>t>>x>>y;
        if (t==1)
        {
            sum=0;
            query (1,1,n,x,y);
            cout<<sum<<"\n";
        }
        if (t==0)                     // primul 1 este pentru nodul cel mai de jos
        {                             // 1 si n sunt capetele grafului
            update2(1,1,n,x,y);       // elementul de pe pozitia x primeste valoarea y
        }
    }
}