Cod sursa(job #2130736)

Utilizator RazvanGutaGuta Razvan Alexandru RazvanGuta Data 13 februarie 2018 21:08:49
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.27 kb
#include<bits/stdc++.h>
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int n,m,x,a,b,poz,val,i,Maxarb[400066],s,sta,fi;
void actualizare(int nod,int st,int dr,int ajuns)
{
    if(st==dr)
    {
        if(ajuns==1)
        Maxarb[nod]=val;
        else
         Maxarb[nod]-=val;
        return;
    }
    int mij=(st+dr)/2;
    if(poz<=mij)
        actualizare(2*nod,st,mij,ajuns);
    else
        actualizare(2*nod+1,mij+1,dr,ajuns);
        Maxarb[nod]=Maxarb[2*nod]+Maxarb[2*nod+1];
}
void interogare(int nod, int st, int dr)
{
     if (sta<=st&&dr<=fi)
     {
          s+=Maxarb[nod];
          return;
     }

     int mij=(st+dr)/2;
     if (sta<=mij)
        interogare(2*nod,st,mij);
     if (mij<fi)
     interogare(2*nod+1,mij+1,dr);
}
int main ()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        poz=i;
        val=x;
        actualizare(1,1,n,1);
    }
    for(i=1;i<=m;i++)
    {
        f>>x>>a>>b;
        if(x==1)
        {
            sta=a;
            fi=b;
            s=0;
            interogare(1,1,n);
            g<<s<<'\n';
        }
        else
        {
            poz=a;
            val=b;
            actualizare(1,1,n,2);
        }
    }
    return 0;
}