Cod sursa(job #2765757)

Utilizator BossBobsterRobert Alexandru Costin BossBobster Data 29 iulie 2021 20:28:50
Problema Heavy Path Decomposition Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 4.44 kb
#include <iostream>
#include <string.h>
#include <random>
#include <fstream>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <iomanip>
#include <algorithm>
#include <math.h>
#include <cmath>
#include <vector>
#include <stack>
#include <queue>
#include <bitset>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <complex>
#include <valarray>
//#include <ext/pb_ds/assoc_container.hpp>
//using namespace __gnu_pbds;
using namespace std;
typedef pair<int, int> pii;
typedef pair<int, string> pis;
typedef pair<string, string> pss;
typedef pair<int, char> pic;
typedef pair<pii, int> piii;
typedef pair<double, double> pdd;
typedef long long ll;
typedef unsigned long long ull;
typedef unsigned int uint;
typedef pair<ll, ll> pll;
typedef pair<int, ll> pil;
typedef pair<ull, ull> pull;
typedef complex<double> cd;
//#define max(n, m) ((n>m)?n:m)
//#define min(n, m) ((n<m)?n:m)
#define f first
#define s second
#define input() ios_base::sync_with_stdio(0);cin.tie(0);

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

int n, m, a, b, cnt = 0, type, lca, bitI = 0;
int nums[100010];
vector<int> adj[100010];
int sz[100010];
int chain[100010];
int pars[100010];
int idx[100010];
int bin[100010][18], dep[200010];
int bitIdx[100010];
int chainSt[100010];
vector<int> tree[100010];
int chainSz[100010];
void up(int which, int idx, int val)
{
    tree[which][idx + chainSz[which]] = val;
    idx += chainSz[which];
    for(int i = idx; i > 1; i >>= 1)
        tree[which][i >> 1] = max(tree[which][i], tree[which][i ^ 1]);
}
int maxQ(int which, int l, int r) // inclusive l, exclusive r
{
    int ans = 0;
    for(l += chainSz[which], r += chainSz[which]; l < r; l >>= 1, r >>= 1)
    {
        if(l & 1)
            ans = max(ans, tree[which][l ++]);
        if(r & 1)
            ans = max(ans, tree[which][-- r]);
    }
    return ans;
}
 
void dfs1(int cur, int par)
{
    pars[cur] = par;
    for(auto it : adj[cur])
    {
        if(it == par) continue;
        dfs1(it, cur);
        sz[cur] += sz[it];
    }
    sz[cur] ++;
}
void dfs2(int cur, int par)
{
    chain[cur] = cnt; chainSz[cnt] ++;
    bitIdx[cur] = bitI++;
    if(adj[cur].size() == 1 && adj[cur][0] == par) return;
    int mx = 100005;
    for(int i = 0; i < adj[cur].size(); i ++)
    {
        if(adj[cur][i] == par) continue;
        if(sz[adj[cur][i]] > sz[mx]) mx = adj[cur][i];
    }
    dfs2(mx, cur);
    for(int i = 0; i < adj[cur].size(); i ++)
    {
        if(adj[cur][i] == par || adj[cur][i] == mx) continue;
        cnt++; chainSt[cnt] = adj[cur][i]; bitI = 0; dfs2(adj[cur][i], cur);
    }
}
void dfsLCA(int cur, int par)
{
    for(int i = 1; i < 18; i ++)
        bin[cur][i] = bin[bin[cur][i-1]][i-1];
    for(auto it : adj[cur])
    {
        if(it == par) continue;
        bin[it][0] = cur, dep[it] = dep[cur]+1;
        dfsLCA(it, cur);
    }
}
int binJump(int cur, int d)
{
    for(int i = 0; i < 18; i ++)
        if((1<<i) & d)
            cur = bin[cur][i];
    return cur;
}
int LCA(int a, int b)
{
    if(dep[a] < dep[b]) swap(a, b);
    a = binJump(a, dep[a]-dep[b]);
    if(a == b) return b;
    for(int i = 17; i >= 0; i --)
        if(bin[a][i] != bin[b][i])
            a = bin[a][i], b = bin[b][i];
    return bin[a][0];
}
int path(int st, int en) //st is up, en is down
{
    int ans = 0;
    if(chain[st] == chain[en]) return maxQ(chain[en], bitIdx[st], bitIdx[en]+1);
    ans = maxQ(chain[en], 0, bitIdx[en]+1);
    en = pars[chainSt[chain[en]]];
    while(chain[st] != chain[en])
    {
        ans = max(ans, maxQ(chain[en], bitIdx[chainSt[chain[en]]], bitIdx[en]+1));
        en = pars[chainSt[chain[en]]];
    }
    ans = max(ans, maxQ(chain[en], bitIdx[st], bitIdx[en]+1));
    return ans;
}
int main()
{
    fin >> n >> m;
    for(int i = 0; i < n; i ++)
        fin >> nums[i];
    for(int i = 0; i < n-1; i ++)
    {
        fin >> a >> b; a--; b--;
        adj[a].push_back(b); adj[b].push_back(a);
    }
    sz[100005] = -1;
    dfs1(0, -1); dfs2(0, -1);
    for(int i = 0; i <= cnt; i ++)
        tree[i].resize(chainSz[i]*2+1);
    for(int i = 0; i < n; i ++)
        up(chain[i], bitIdx[i], nums[i]);
    dfsLCA(0, -1);
    while(m--)
    {
        fin >> type >> a >> b;
        if(type == 0)
        {
            a--;
            up(chain[a], bitIdx[a], b);
            nums[a] = b;
        }
        else
        {
            a--; b--;
            lca = LCA(a, b);
            fout << max(path(lca, a), path(lca, b)) << "\n";
        }
    }
    return 0;
}