Codevs 3305 水果姐逛水果街Ⅱ

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

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int M=200000+50;
const int inf=1000000000+50;
vector<int> tu[M];
int ans1[M][25],ans2[M][25],maxx[M][25],minn[M][25],deep[M],fa[M][25],a[M];
int n,m,u,v,z;
void build(int f,int t){
    tu[f].push_back(t);
    return ;
}
void dfs(int t,int f){
    deep[t]=deep[f]+1;
    fa[t][0]=f;
    maxx[t][0]=max(a[t],a[f]);
    minn[t][0]=min(a[t],a[f]);
    ans1[t][0]=max(0,a[f]-a[t]);
    ans2[t][0]=max(0,a[t]-a[f]);
    for(int i=1;i<=20;i++){//printf("hha");
        fa[t][i]=fa[fa[t][i-1]][i-1];
        maxx[t][i]=max(maxx[t][i-1],maxx[fa[t][i-1]][i-1]);
        minn[t][i]=min(minn[t][i-1],minn[fa[t][i-1]][i-1]);
        ans1[t][i]=max(max(ans1[t][i-1],ans1[fa[t][i-1]][i-1]),maxx[fa[t][i-1]][i-1]-minn[t][i-1]);
        ans2[t][i]=max(max(ans2[t][i-1],ans2[fa[t][i-1]][i-1]),maxx[t][i-1]-minn[fa[t][i-1]][i-1]);
    }
    for(int i=0;i<tu[t].size();i++){
        int u=tu[t][i];
        if(u!=f){
            dfs(u,t);
        }
    }
    return ;
}
int find_lca(int x,int y){
    if(deep[x]<deep[y]) swap(x,y);
    if(deep[x]>deep[y]) {
        int d=deep[x]-deep[y];
        for(int i=0;i<=20;i++){
            if(d>>i&1) x=fa[x][i];
        }
    }
    if(x==y) return x;
    else{
        for(int i=20;i>=0;i--){
            if(fa[x][i]!=fa[y][i]) 
                x=fa[x][i],y=fa[y][i];
        }
        return fa[x][0];
    }
}
int done(int x,int y){
    int lca=find_lca(x,y);
    int ans=0,maxn=-inf,mixn=inf;
    int d=deep[x]-deep[lca];
    for(int i=20;i>=0;i--){
        if(d>>i&1){
            ans=max(ans,max(ans1[x][i],maxx[x][i]-mixn));
            mixn=min(mixn,minn[x][i]);
            x=fa[x][i];
        }
    }
     d=deep[y]-deep[lca];
    for(int i=20;i>=0;i--){
        if(d>>i&1){
            ans=max(ans,max(ans2[y][i],maxn-minn[y][i]));
            maxn=max(maxn,maxx[y][i]);
            y=fa[y][i];
        }
    }
    return max(ans,maxn-mixn);
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++) {
        scanf("%d%d",&u,&v);
        build(u,v);build(v,u);
    }

    dfs(1,1);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        printf("%d\n",done(u,v));
    }
}