Cod sursa(job #2761119)

Utilizator Virgil993Virgil Turcu Virgil993 Data 30 iunie 2021 17:52:11
Problema Arbori de intervale Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.7 kb
#include <iostream>
#include<bits/stdc++.h>

using namespace std;

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


int arbore[4*100001];
int v[100001],n;

void creaza_arbore(int indx,int st,int dr) //creez arobrele de intervale
{
    if(st==dr)
    {
        arbore[indx] = v[st]; //daca am ajuns la un interval cu o singura valoare atunci punem valoarea din vector in arbore pe nodul respectiv
    }
    else
    {
        int mij = (st+dr)/2;
        creaza_arbore(indx*2,st,mij);//daca intervalul pe care suntem nu este de lungime 1, creeam subarborele drept si cel stang si la final actualizam valoarea din nodul curent cu val maxima
        //din cei doi subarbori
        creaza_arbore(indx*2+1,mij+1,dr);
        arbore[indx] = max(arbore[indx*2],arbore[indx*2+1]);
    }
}

int maxim(int indx,int st,int dr,int start,int finish)
{
    if(start<=st&&finish>=dr)
    {
        return arbore[indx];//daca intervalul curent se regaseste in in interiorul intervalului cautat atunci doar returnam valoarea
    }
    else
    {
        int mij = (st+dr)/2;
        int m1=-1,m2=-1;
        if(start<=mij)//daca pozitia de inceput este mai mica decat mijlocul intervalului curent atunci ne ducem in stanga deoarece trebuie sa verificam o anumita bucata din subarborele drept
            m1 = maxim(indx*2,st,mij,start,finish);
        if(finish>mij)//daca pozitia de final este mai mare strict decat mijlocul intervalului curent, ne ducem in dreapta deoarece trebuie sa verificam o anumita bucata din subarborele drept
            m2 = maxim(indx*2+1,mij+1,dr,start,finish);
        return max(m1,m2);//la final comparam cele doua valori obtinute in urma interogarii
    }
}

void insereaza(int indx,int st,int dr,int val,int poz)
{
    if(st==dr)
    {
        arbore[indx] = val;//daca suntem pe un interval de lungime 1 inseamna ca am ajuns la pozitia dorita si doar schimbam valoarea nodului curent
    }
    else
    {
        int mij = (st+dr)/2;
        if(poz<=mij)//daca pozitia este mai mica ca mijlocul, inseamna ca trb sa cautam in subarborele drept si vice versa daca pozitia este mai mare ca mijlocul
            insereaza(indx*2,st,mij,val,poz);
        else
            insereaza(indx*2+1,mij+1,dr,val,poz);
        arbore[indx] = max(arbore[indx*2],arbore[indx*2+1]);//la final acutalizam nodurile cu noile valori maxima dupa inserare.
    }

}

int main()
{
    int m,x,y,z;
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        fin>>v[i];

    creaza_arbore(1,1,n);
    for(int i=0;i<m;i++)
    {

        fin>>x>>y>>z;
        if(x==0)
            fout<<maxim(1,1,n,y,z)<<"\n";
        else
            insereaza(1,1,n,z,y);


    }

    return 0;
}