Cod sursa(job #1678019)

Utilizator bogdanboboc97Bogdan Boboc bogdanboboc97 Data 6 aprilie 2016 22:29:43
Problema Heavy Path Decomposition Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 3.43 kb
#include <iostream>
#include <vector>
#include <algorithm>
#include <limits>
#include <numeric>
#include <cstring>
#include <string>
#include <queue>
#include <set>
#include <cmath>
#include <fstream>
#include <cstdlib>
#include <map>
#include <stack>
#define pb push_back
#define mp make_pair
#define INF numeric_limits<int>::max()
#define lsb(x) (-x)&x
#define int64 long long
using namespace std;
ifstream in("heavypath.in");
ofstream out("heavypath.out");
vector< vector<int> > adj;
vector<int> v,eul,lvl,pos,firstNode,comp,parent,pathPos,tree;
priority_queue< pair<int,int> > leafs;
int n,k,np;
void euler(int x,int px)
{
    lvl[x]=lvl[px]+1;
    eul[++k]=x;
    pos[x]=k;
    parent[x]=px;
    if(adj[x].size()==1 && x!=1)
        leafs.push(mp(lvl[x],x));
    else for(auto it: adj[x])
    if(it!=px)
    {
        euler(it,x);
        eul[++k]=x;
    }
}
int RMQ[18][200020],lg[200020];
void rmq()
{
    for(int i=1;i<=k;i++)
    {
        if(i>=2)
            lg[i]=lg[i/2]+1;
        RMQ[0][i]=eul[i];
    }
    for(int i=1;(1<<i)<=k;i++)
    for(int j=1;j<=k-(1<<i)+1;j++)
        RMQ[i][j]=( lvl[ RMQ[i-1][j] ] > lvl[ RMQ[i-1][j+(1<<(i-1))] ] ? RMQ[i-1][j+(1<<(i-1))] : RMQ[i-1][j] );
}
int LCA(int x,int y)
{
    x=pos[x];
    y=pos[y];
    if(x>y)
        swap(x,y);
    int diff=y-x+1;
    return (lvl[ RMQ[lg[diff]][x] ]  > lvl[ RMQ[lg[diff]][x+diff-(1<<lg[diff])] ] ? RMQ[lg[diff]][x+diff-(1<<lg[diff])] : RMQ[lg[diff]][x]);
}
void pathUpd(int x,int left,int right,int position,int val)
{
    if(left==right)
        tree[x]=val;
    else{
        int mid=(left+right)/2;
        if(position<=mid)
            pathUpd(2*x,left,mid,position,val);
        else
            pathUpd(2*x+1,mid+1,right,position,val);
        tree[x]=max(tree[2*x],tree[2*x+1]);
    }
}
int sol;
void pathQuery(int x,int left,int right,int a,int b)
{
    if(a<=left && b>=right)
        sol=max(sol,tree[x]);
    else{
        int mid=(left+right)/2;
        if(a<=mid)
            pathQuery(2*x,left,mid,a,b);
        if(b>mid)
            pathQuery(2*x+1,mid+1,right,a,b);
    }
}
void update(int x,int val)
{
    pathUpd(1,1,n,pathPos[x],val);
}
void getMax(int x,int lca)
{
    int node=x;
    while(comp[node]!=comp[lca])
    {
        pathQuery(1,1,n,pathPos[ firstNode[ comp[node] ] ],pathPos[node]);
        node=parent[ firstNode[ comp[node] ] ];
    }
    pathQuery(1,1,n,pos[lca],pathPos[node]);
}
int query(int x,int y)
{
    int lca=LCA(x,y);
    sol=-INF;
    getMax(x,lca);
    getMax(y,lca);
    return sol;
}
int main()
{
    int q;
    in>>n>>q;
    adj=vector< vector<int> > (n+1);
    lvl=pos=parent=firstNode=comp=pathPos=v=vector<int> (n+1);
    eul=vector<int> (2*n+3);
    tree=vector<int> (4*n+1);
    for(int i=1;i<=n;i++)
        in>>v[i];
    for(int i=2;i<=n;i++)
    {
        int x,y;
        in>>x>>y;
        adj[x].pb(y);
        adj[y].pb(x);
    }
    euler(1,0);
    comp[0]=-1;
    lvl[0]=INF;
    rmq();
    int t=0;
    while(!leafs.empty())
    {
        np++;
        int x,nr=0;
        for(x=leafs.top().second;comp[x]==0;x=parent[x])
            nr++;
        for(int i=nr,x=leafs.top().second;comp[x]==0;x=parent[x],i--)
        {
            pathUpd(1,1,n,i+t,v[x]);
            comp[x]=np;
            firstNode[np]=x;
            pathPos[x]=i+t;
        }
        t+=nr;
        leafs.pop();
    }
    for(;q;q--)
    {
        int x,y,t;
        in>>t>>x>>y;
        if(t==0)
            update(x,y);
        else
            out<<query(x,y)<<'\n';
    }
    return 0;
}