这个题卡着时间过的,先留下代码,等过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;
}