poj3728 倍增法lca 好题!

lca的好题!网上用st表和离线解的比较多,用树上倍增也是可以做的

不知道错在哪里,等刷完了这个专题再回来看

题解链接https://blog.csdn.net/Sd_Invol/article/details/9572423

/*
给一颗点权树,求出一个点对(x,y)之间的max{A,B,C}
A:x到lca路径上的最大差值
B:lca到y路径上的最大差值
C:x到y路径上的最大差值
需要维护的值,x结点到的祖先,x结点到祖先路径上的最大值,最小值,x结点到路径上的最大收益,最小收益(可以是负数)
对于每个询问x->y:A:x到祖先的最大收益
B:y到祖先的最小收益的负值
C:x到lca的最大值减去lca到y的最小值
*/
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
#define maxn 50005
struct Edge{
int x,next;
}e[maxn<<];
int head[maxn],tot,n,q;
inline void addedge(int u,int v){
e[tot].x=v;
e[tot].next=head[u];
head[u]=tot++;
}
int d[maxn],f[maxn][],a[maxn];//深度,祖先,点权
int fm[maxn][],fn[maxn][],sm[maxn][],sn[maxn][];//最大值,最小值,最大收益,最小收益
void init(){
tot=;
memset(head,-,sizeof head);
}
void dfs(int x,int fa,int dep){
d[x]=dep,f[x][]=fa;
for(int i=head[x];i!=-;i=e[i].next)
if(fa!=e[i].x) dfs(e[i].x,x,dep+);
}
int query(int x,int y){
int i,xx=,yy=,X=a[x],Y=a[y];//x侧最大收益,y侧最小收益,x侧最小值,y侧最大值
i=;
while(d[x]!=d[y]){//拉倒同一高度
if(abs(d[x]-d[y]) >= <<i)
if(d[y]<d[x])
xx=max(max(xx,sm[x][i]),fm[x][i]-X),X=min(X,fn[x][i]),x=f[x][i];
else
yy=min(min(yy,sn[y][i]),fn[y][i]-Y),Y=max(Y,fm[y][i]),y=f[y][i];
--i;
}
if(x==y) return max(max(xx,-yy),Y-X);
i=;
while(i>=){
if(f[x][i] && f[y][i] && f[x][i]!=f[y][i]){
xx=max(max(xx,sm[x][i]),fm[x][i]-X),X=min(X,fn[x][i]),x=f[x][i];
yy=min(min(yy,sn[y][i]),fn[y][i]-Y),Y=max(Y,fm[y][i]),y=f[y][i];
}
--i;
}
i=;//这里还要跳一次
xx=max(max(xx,sm[x][i]),fm[x][i]-X),X=min(X,fn[x][i]),x=f[x][i];
yy=min(min(yy,sn[y][i]),fn[y][i]-Y),Y=max(Y,fm[y][i]),y=f[y][i];
return max(max(xx,-yy),Y-X);
}
void work(){
int x,y;
init();
scanf("%d",&n);
for(int i=;i<=n;i++) scanf("%d",&a[i]);
for(int i=;i<n;i++){
scanf("%d%d",&x,&y);
addedge(x,y);addedge(y,x);
}
dfs(,,);
fm[][]=fn[][]=a[];
sm[][]=-<<;sn[][]=<<;
for(int i=;i<=n;i++){//先打初始状态
fm[i][]=max(a[i],a[f[i][]]);//和父亲比较
fn[i][]=min(a[i],a[f[i][]]);
sm[i][]=max(,a[f[i][]]-a[i]);//要么是0,要么赚了钱
sn[i][]=min(,a[f[i][]]-a[i]);//要么是0,要么是亏了钱
}
for(int j=;(<<j)<n;j++)//再打剩下状态
for(int i=;i<=n;i++){
int tmp=f[i][j-];//中间态
f[i][j]=f[tmp][j-];
fm[i][j]=max(fm[i][j-],fm[tmp][j-]);
fn[i][j]=min(fm[i][j-],fm[tmp][j-]);
sm[i][j]=max(max(sm[i][j-],sm[tmp][j-]),fm[tmp][j-]-fn[i][j-]);//最大收益要么是两段间的最大收益,要么是祖先段的最大收益减去子孙段的最小收益
sn[i][j]=min(min(sn[i][j-],sn[tmp][j-]),fn[tmp][j-]-fm[i][j-]);//最小收益相反
}
scanf("%d",&q);
while(q--){
scanf("%d%d",&x,&y);
printf("%d\n",query(x,y));
}
}
int main(){
work();
return ;
}
上一篇:【POJ】2348 Euclid's Game(扩欧)


下一篇:【BZOJ2154】Crash的数字表格(莫比乌斯反演)