POJ-3728-The merchant

這個題卡着時間過的，先留下代碼，等過2天再重寫試試。

代碼：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
const int maxn=6e4+100;
const int maxm=maxn*2;
struct Query
{
    int v;
    int id;
    int pos;
    Query(){}
    Query(int sv,int sid,int spos):v(sv),id(sid),pos(spos){}
};
struct Node{
    int down;
    int up;
    int highpri;
    int lowpri;
    int pre;
}a[maxn];
int e,head[maxn],pnt[maxm],nxt[maxm];
int n,k,pri[maxn],p[maxn],ans[maxn];
bool vis[maxn];
vector<Query> querys[maxn];
void AddEdge(int u,int v)
{
    pnt[e]=v;nxt[e]=head[u];head[u]=e++;
}
int find(int x)
{
    if(p[x]==x)
        return x;
    int f=p[x];
    p[x]=find(p[x]);
    a[x].up=max(a[f].up,max(a[x].up,a[x].highpri-a[f].lowpri));
    a[x].down=max(a[f].down,max(a[x].down,a[f].highpri-a[x].lowpri));
    a[x].lowpri=min(a[x].lowpri,a[f].lowpri);
    a[x].highpri=max(a[x].highpri,a[f].highpri);
    return p[x];
}
void LCA(int u,int fa)
{
    p[u]=u;
    for(int i=head[u];i!=-1;i=nxt[i])
    {
        if(pnt[i]==fa)
            continue;
        a[pnt[i]].pre=u;
        LCA(pnt[i],u);
        p[find(pnt[i])]=find(u);
    }
    vis[u]=1;
    for(int i=0;i<querys[u].size();i++)
    {
        if(vis[querys[u][i].v])
        {
            int f=find(querys[u][i].v);
            int v,mini,maxi,val=0;
            v=u;
            maxi=mini=pri[u];
            if(querys[u][i].pos==0)
            {
                while(v!=f)
                {
                    val=max(a[v].up,max(val,a[v].highpri-mini));
                    mini=min(mini,a[v].lowpri);
                    v=a[v].pre;
                }
                ans[querys[u][i].id]=max(max(val,a[querys[u][i].v].up),a[querys[u][i].v].highpri-mini);
            }
            else
            {
                while(v!=f)
                {
                    val=max(a[v].down,max(val,maxi-a[v].lowpri));
                    maxi=max(maxi,a[v].highpri);
                    v=a[v].pre;
                }
                ans[querys[u][i].id]=max(max(val,a[querys[u][i].v].down),maxi-a[querys[u][i].v].lowpri);
            }
        }
    }
}
void Init()
{
    for(int i=1;i<=n;i++)
        querys[i].clear();
    e=0;
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++)
    {
        a[i].lowpri=a[i].highpri=pri[i];
        a[i].up=a[i].down=0;
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&pri[i]);
        Init();
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            AddEdge(u,v);
            AddEdge(v,u);
        }
        scanf("%d",&k);
        for(int i=0;i<k;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            querys[u].push_back(Query(v,i,0));
            querys[v].push_back(Query(u,i,1));
        }
        LCA(1,-1);
        for(int i=0;i<k;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}