【树上倍增】【分类讨论】水果姐逛水果街Ⅱ

3305 水果姐逛水果街Ⅱ

 时间限制: 2 s
 空间限制: 256000 KB
 题目等级 : 钻石 Diamond

题目描述 Description

水果姐第二天心情也很不错，又来逛水果街。
突然，cgh又出现了。cgh施展了魔法，水果街变成了树结构（店与店之间只有一条唯一的路径）。
同样还是n家水果店，编号为1~n，每家店能买水果也能卖水果，并且同一家店卖与买的价格一样。
cgh给出m个问题，每个问题要求水果姐从第x家店出发到第y家店，途中只能选一家店买一个水果，然后选一家店（可以是同一家店，但不能往回走）卖出去。求最多可以赚多少钱。
水果姐向学过oi的你求助。

输入描述 Input Description

第一行n，表示有n家店
下来n个正整数，表示每家店一个苹果的价格。
下来n-1行，每行两个整数x，y，表示第x家店和第y家店有一条边。
下来一个整数m，表示下来有m个询问。
下来有m行，每行两个整数x和y，表示从第x家店出发到第y家店。

输出描述 Output Description

有m行。
每行对应一个询问，一个整数，表示面对cgh的每次询问，水果姐最多可以赚到多少钱。

样例输入 Sample Input

10
16 5 1 15 15 1 8 9 9 15 
1 2
1 3
2 4
2 5
2 6
6 7
4 8
1 9
1 10
6
9 1
5 1
1 7
3 3
1 1
3 6

样例输出 Sample Output

7
11
7
0
0
15

数据范围及提示 Data Size & Hint

0<=苹果的价格<=10^8
0<n<=200000
0<m<=10000

题解：

树上查询没有修改，为了图简单就写了树上倍增，结果发现写起来也是有些蛋疼的。。在写dfs的时候一定要先处理父亲在继续搜啊（为此调了30min+）。。

Code：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 1000010
#define F fa[x][i-1]
using namespace std;
struct point{
    int v,next;
}edge[2*N];
int n,m,num=0,maxn,minn,ans;
int head[N],a[N],de[N],fa[N][20],ma[N][20],mi[N][20],cq[N][20],ch[N][20];
bool vis[N]={0};
int in(){
    int x=0; char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
void add(int u,int v){
    edge[++num].v=v; edge[num].next=head[u]; head[u]=num;
}
int lca(int x,int y){
    if (de[x]<de[y]) swap(x,y);
    int k=de[x]-de[y];
    for (int i=0; i<=18; i++)
        if (k&(1<<i)) x=fa[x][i];
    for (int i=18; i>=0; i--)
        if (fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    if (x==y) return x;
    return fa[x][0];
}
void dfs(int x){
    vis[x]=true;
    for (int i=1; i<=18; i++){
        if (de[x]<(1<<i)) break;
        fa[x][i]=fa[F][i-1];
        ma[x][i]=max(ma[x][i-1],ma[F][i-1]);
        mi[x][i]=min(mi[x][i-1],mi[F][i-1]);
        cq[x][i]=max(max(cq[x][i-1],cq[F][i-1]),ma[F][i-1]-mi[x][i-1]);
        ch[x][i]=max(max(ch[x][i-1],ch[F][i-1]),ma[x][i-1]-mi[F][i-1]);
    }
    for (int i=head[x]; i; i=edge[i].next){
        int v=edge[i].v;
        if (vis[v]) continue;
        fa[v][0]=x; de[v]=de[x]+1;
        ma[v][0]=max(a[x],a[v]); mi[v][0]=min(a[x],a[v]);
        cq[v][0]=a[x]-a[v]; ch[v][0]=a[v]-a[x];
        dfs(v);
    }
}
void work(int x,int y){
    maxn=0,minn=0x7fffffff,ans=0;
    int ff=lca(x,y),k;
//  cout<<x<<"  "<<y<<"  "<<ff<<"  "<<endl;
    k=de[x]-de[ff];
//  cout<<k<<endl;
    if (k>0){
        for (int i=0; i<=18; i++)
            if (k&(1<<i)){
                ans=max(ans,max(cq[x][i],ma[x][i]-minn));
                minn=min(minn,mi[x][i]); x=fa[x][i];
//              cout<<ans<<"   "<<minn<<endl;
            }
    }
    k=de[y]-de[ff];
//  cout<<k<<endl;
    if (k>0){
        for (int i=0; i<=18; i++)
            if (k&(1<<i)){
                ans=max(ans,max(ch[y][i],maxn-mi[y][i]));
                maxn=max(maxn,ma[y][i]); y=fa[y][i];
//              cout<<ans<<"   "<<maxn<<endl;
            }
    }
    ans=max(ans,maxn-minn);
}
int main(){
    n=in();
    for (int i=1; i<=n; i++) a[i]=in();
    for (int i=1; i<n; i++){
        int u=in(),v=in();
        add(u,v),add(v,u);
    }
    fa[1][0]=1; dfs(1);
    m=in();
    while (m--){
        int x=in(),y=in();
        work(x,y);
        printf("%d\n",ans);
    }
    return 0;
}