Cod sursa(job #1952326)

Utilizator amaliarebAmalia Rebegea amaliareb Data 4 aprilie 2017 01:41:09
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <cstring>

using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int n,i,j,v[20000],aint[262145],val,a,b,m,x,sum,poz;

void update(int nod, int st, int dr)
{
    int mid=(st+dr)/2;
    aint[nod]+=val;
    if(st==dr) return;
    if(poz<=mid) update(2*nod,st,mid);
    else update(2*nod+1,mid+1,dr);
}

void query(int nod, int st, int dr)
{
    int mid=(st+dr)/2;
    if(a<=st && b>=dr)
    {
        sum+=aint[nod];
        return;
    }
    if(a<=mid) query(2*nod,st,mid);
    if(b>mid) query(2*nod+1,mid+1,dr);
}

int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>v[i];
        poz=i;
        val=v[i];
        update(1,1,n);
    }
    for(j=1;j<=m;j++)
    {
        f>>x;
        if(x==0)
        {
            f>>poz>>val;
            val=-val;
            update(1,1,n);
        }
        else
        {
            f>>a>>b;
            sum=0;
            query(1,1,n);
            g<<sum<<'\n';
        }
    }
    return 0;
}