Cod sursa(job #1678482)

Utilizator bogdanboboc97Bogdan Boboc bogdanboboc97 Data 7 aprilie 2016 12:47:00
Problema Heavy Path Decomposition Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 2.87 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> lvl,v,firstNode,comp,parent,pos,tree;
priority_queue< pair<int,int> > leafs;
int n,k,np;
void dfs(int x,int px)
{
    lvl[x]=lvl[px]+1;
    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)
        dfs(it,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,pos[x],val);
}
int query(int x,int y)
{
    sol=-INF;
    while(comp[x]!=comp[y])
    {
        if(lvl[ firstNode[ comp[x] ] ]>lvl[ firstNode[ comp[y] ] ])
        {
            pathQuery(1,1,n,pos[ firstNode[ comp[x] ] ],pos[x]);
            x=parent[ firstNode[ comp[x] ] ];
        }
        else{
            pathQuery(1,1,n,pos[ firstNode[ comp[y] ] ],pos[y]);
            y=parent[ firstNode[ comp[y] ] ];
        }
    }
    if(pos[x]>pos[y])
        swap(x,y);
    pathQuery(1,1,n,pos[x],pos[y]);
    return sol;
}
int main()
{
    int q;
    in>>n>>q;
    adj=vector< vector<int> > (n+1);
    lvl=parent=firstNode=comp=pos=v=vector<int> (n+1);
    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);
    }
    dfs(1,0);
    comp[0]=-1;
    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;
            pos[x]=i+t;
        }
        t+=nr;
        leafs.pop();
    }
    parent[1]=1;
    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;
}