Cod sursa(job #2408198)

Utilizator smoc_georgemarianSmoc George-Marian smoc_georgemarian Data 17 aprilie 2019 18:28:33
Problema Arbori de intervale Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.64 kb
#include <bits/stdc++.h>
#define NMAX 400009
using namespace std;
ifstream fin("arbint.in");
ofstream fout("arbint.out");
int a[NMAX];
int lzy[NMAX];
int v[NMAX/4];
int n,m;
void build(int nod, int st, int dr);
int query(int nod, int st, int dr, int x, int y);
void update(int nod, int st, int dr, int x, int y, int val);
void citire();
int main()
{int c,x,y;
   citire();
   build(1,1,n);
   for(int i=1;i<=m;i++)
    {
        fin>>c>>x>>y;
        if(c==0)
           fout<<query(1,1,n,x,y)<<'\n';
           else
           update(1,1,n,x,x,y-v[x]);
    }
    return 0;
}
void citire()
{
 int i;
 fin>>n>>m;
 for(i=1;i<=n;i++)
        fin>>v[i];
}
void build(int nod, int st, int dr)
{
 if(st==dr)
        {a[nod]=v[st];return;}
 int mj=(st+dr)/2;
 build(2*nod,st,mj);
 build(2*nod+1,mj+1,dr);
 a[nod]=max(a[2*nod],a[2*nod+1]);
}
void update(int nod, int st, int dr, int x, int y, int val)
{
  if(lzy[nod])
    {
     if(st!=dr)
        {
         lzy[2*nod+1]+=lzy[nod];
         lzy[2*nod]+=lzy[nod];
         a[2*nod]+=lzy[nod];
         a[2*nod+1]+=lzy[nod];

        }
      lzy[nod]=0;
    }
   if(x<=st && dr<=y)
        {a[nod]+=val,lzy[nod]+=val;return;}
    int mj=(st+dr)/2;
    if(x<=mj)
         update(2*nod,st,mj,x,y,val);
    if(y>mj)
        update(2*nod+1,mj+1,dr,x,y,val);
    a[nod]=max(a[2*nod+1],a[2*nod]);

}
int query(int nod, int st, int dr, int x, int y)
{
   int ls=0,ld=0;
   int mj;
   if(x<=st && dr<=y)
      return a[nod];
   mj=(st+dr)/2;
   if(x<=mj)
       ls=query(2*nod,st,mj,x,y);
   if(y>mj)
        ld=query(2*nod+1,mj+1,dr,x,y);
   return max(ls,ld);
}