Cod sursa(job #2064935)

Utilizator stelian2000Stelian Chichirim stelian2000 Data 13 noiembrie 2017 00:45:00
Problema Heavy Path Decomposition Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.7 kb
#include <cstdio>
#include <vector>
#include <algorithm>

using namespace std;

int cnt[100010],lant[100010],size_lant[100010],poz_lant[100010],tata_lant[100010],niv[100010],nr_lant,val[100010];
vector<int> v[100010],arb[100010];

void update(int nod,int st,int dr,int poz,int val,vector<int> &arb)
{
    if(st==dr) arb[nod]=val;
    else
    {
        int mid=(st+dr)/2;
        if(poz<=mid) update(nod*2,st,mid,poz,val,arb);
        else update(nod*2+1,mid+1,dr,poz,val,arb);
        arb[nod]=max(arb[nod*2],arb[nod*2+1]);
    }
}

int query(int nod,int st,int dr,int left,int right,vector<int> &arb)
{
    if(left<=st && dr<=right) return arb[nod];
    int mid=(st+dr)/2,val=0;
    if(left<=mid) val=query(nod*2,st,mid,left,right,arb);
    if(mid<right) val=max(val,query(nod*2+1,mid+1,dr,left,right,arb));
    return val;
}

void dfs1(int nod,int tata,int nivel)
{
    cnt[nod]=1;
    niv[nod]=nivel;
    for(int i=0;i<v[nod].size();i++)
        if(v[nod][i]!=tata)
        {
            dfs1(v[nod][i],nod,nivel+1);
            cnt[nod]+=cnt[v[nod][i]];
        }
}

void dfs2(int nod,int tata,int l)
{
    lant[nod]=l;
    size_lant[l]++;
    poz_lant[nod]=size_lant[l];
    int heavy=0;
    for(int i=0;i<v[nod].size();i++)
        if(v[nod][i]!=tata && cnt[heavy]<cnt[v[nod][i]]) heavy=v[nod][i];
    if(heavy==0) return;
    dfs2(heavy,nod,l);
    for(int i=0;i<v[nod].size();i++)
        if(v[nod][i]!=heavy && v[nod][i]!=tata)
        {
            nr_lant++;
            tata_lant[nr_lant]=nod;
            dfs2(v[nod][i],nod,nr_lant);
        }
}

int solve(int a,int b)
{
    int sol=0;
    while(lant[a]!=lant[b])
    {
        if(niv[tata_lant[lant[a]]]<niv[tata_lant[lant[b]]]) swap(a,b);
        sol=max(sol,query(1,1,size_lant[lant[a]],1,poz_lant[a],arb[lant[a]]));
        a=tata_lant[lant[a]];
    }
    if(niv[a]>niv[b]) swap(a,b);
    sol=max(sol,query(1,1,size_lant[lant[a]],poz_lant[a],poz_lant[b],arb[lant[a]]));
    return sol;
}

int main()
{
    freopen("heavypath.in","r",stdin);
    freopen("heavypath.out","w",stdout);
    int n,m,x,y,t;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&val[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs1(1,0,1);
    nr_lant=1;
    dfs2(1,0,1);
    for(int i=1;i<=nr_lant;i++) arb[i].resize(size_lant[i]*4);
    for(int i=1;i<=n;i++) update(1,1,size_lant[lant[i]],poz_lant[i],val[i],arb[lant[i]]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&t,&x,&y);
        if(t==0) update(1,1,size_lant[lant[x]],poz_lant[x],y,arb[lant[x]]);
        else printf("%d\n",solve(x,y));
    }
    return 0;
}