Cod sursa(job #3226771)

Utilizator AndreiBOTOBotocan Andrei AndreiBOTO Data 22 aprilie 2024 19:24:45
Problema Heavy Path Decomposition Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.01 kb
#include <bits/stdc++.h>

#pragma optimize GCC ("Ofast")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

///#include <tryhardmode>
///#include <GODMODE::ON>

using namespace std;

#define cin fin
#define cout fout

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

const int NMAX=2e5+5;

vector<int>v[NMAX];
int aint[4*NMAX];
int head[NMAX];
int sarb[NMAX];
int lvl[NMAX];
int ti[NMAX];
int a[NMAX];
int t[NMAX];

int kon;
int n;

void dfs1(int p,int tata)
{
    lvl[p]=lvl[tata]+1;
    head[p]=p;
    sarb[p]=1;
    t[p]=tata;
    for(auto i:v[p])
    {
        if(i!=tata)
        {
            dfs1(i,p);
            sarb[i]+=sarb[p];
        }
    }
}

void dfs2(int p)
{
    int maxim=-1,hc=-1;
    ti[p]=++kon;
    for(auto i:v[p])
    {
        if(i!=t[p])
        {
            if(maxim<sarb[i])
            {
                maxim=sarb[i];
                hc=i;
            }
        }
    }
    if(hc<0)
        return ;
    else
    {
        head[hc]=head[p];
        dfs2(hc);
        for(auto i:v[p])
        {
            if(i!=t[p] && i!=hc)
                dfs2(i);
        }
    }
}

void build(int p,int st,int dr)
{
    int mij;
    if(st==dr)
    {
        aint[p]=a[ti[st]];
        return ;
    }
    else
    {
        mij=st+dr;
        mij/=2;
        build(2*p,st,mij);
        build(2*p+1,mij+1,dr);
        aint[p]=max(aint[2*p],aint[2*p+1]);
    }
}

void update(int p,int st,int dr,int a,int b)
{
    int mij;
    if(st==dr)
    {
        aint[p]=b;
        return ;
    }
    else
    {
        mij=st+dr;
        mij/=2;
        if(a<=mij)
            update(2*p,st,mij,a,b);
        else
            update(2*p+1,mij+1,dr,a,b);
        aint[p]=max(aint[2*p],aint[2*p+1]);
    }
}

int query(int p,int st,int dr,int a,int b)
{
    if(a<=st && dr<=b)
        return aint[p];
    else
    {
        int maxi=-1;
        int mij=st+dr;
        mij/=2;
        if(a<=mij)
            maxi=max(maxi,query(2*p,st,mij,a,b));
        if(mij<b)
            maxi=max(maxi,query(2*p+1,mij+1,dr,a,b));
        return maxi;
    }
}

int solve(int a,int b)
{
    if(head[a]==head[b])
    {
        if(ti[a]>ti[b])
            swap(a,b);
        return query(1,1,n,ti[a],ti[b]);
    }
    if(lvl[head[a]]<lvl[head[b]])
        swap(a,b);
    int aux=solve(t[head[a]],b);
    return max(aux,query(1,1,n,ti[head[a]],ti[a]));
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int m,i,j,q;
    cin>>n>>q;
    for(i=1;i<=n;i++)
        cin>>a[i];
    for(i=1;i<n;i++)
    {
        int x,y;
        cin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs1(1,0);
    dfs2(1);
    build(1,1,n);
    while(q--)
    {
        int type,x,y;
        cin>>type>>x>>y;
        if(type==0)
            update(1,1,n,ti[x],y);
        if(type==1)
            cout<<solve(x,y)<<"\n";
    }
    return 0;
}