LCA倍增（zstu4395）

题目链接：http://oj.acm.zstu.edu.cn/JudgeOnline/problem.php?id=4395

思路：倍增找公共祖先，顺便处理出路径上最大值。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define pb push_back
#define lc (d<<1)
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
const int N=20008;
int deep[N],fa[N][20],n,m,e[N][20],s[N],p[N],mx;
bool vis[N];
vector<int> g[N];
void dfs(int u)
{
    vis[u]=1;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(!vis[v])   deep[v]=deep[u]+1,dfs(v),p[v]=u;
    }
}
void bz()
{
    FOR(i,1,18)
     FOR(j,1,n)
      fa[j][i]=fa[fa[j][i-1]][i-1],e[j][i]=max(e[j][i-1],e[fa[j][i-1]][i-1]);
}
int lca(int x,int y)
{
    mx=0;
    if(deep[x]<deep[y])  swap(x,y);
    int d=deep[x]-deep[y];
    mx=max(e[x][0],e[y][0]);
    FOR(i,0,18)
     if(d&(1<<i))  mx=max(mx,e[x][i]),x=fa[x][i];
    if(x==y)  return x;
    FOL(i,18,0)
     if(fa[x][i]!=fa[y][i])  mx=max(mx,e[x][i]),mx=max(mx,e[y][i]),x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void ans(int x,int y)
{
    int t=lca(x,y);
    if(t!=x&&t!=y)  {puts("-1");return;}
    cout<<mx<<endl;
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m;
    int a,b,z;
    FOR(i,1,n-1)
    {
        si(a),si(b);
        g[b].pb(a);
        g[a].pb(b);
    }
    FOR(i,1,n)  si(e[i][0]);
    dfs(1);
    FOR(i,1,n)  if(p[i])  fa[i][0]=p[i];
    bz();
    while(m--)
    {
        si(a),si(b);
        ans(a,b);
    }
    return 0;
}