bzoj2870 最长道路

传送门

参考博客

记录点权最小值为$M$时的最长路径$L$。（以对应的分治中心为根）
令当前点到分治中心最小值为$min$。那么我们只需要找到$M\ge min$对应的$L$的最大值。那么答案就是$M\ast (L+当前L)$。

对于这个最大值可以用树状数组维护。下标表示点权。
$query(x)$返回最小点权为$x$的最长路径长度。
$update(x,l)$是用最小点权为$x$，长度为$l$的这个点到根的路径去更新树状数组。
搜完一个子树更新完答案后，再更新树状数组。

由于更新答案的时候，最小值$M$是固定的，只会用之前遍历过的子树去更新，没有算及后面子树可能更新出更优答案。所以需要顺着反着做两次。

一次分治完之后，为了避免$memset$树状数组带来的巨大消耗，用一个$now$变量表示时间。如果某节点的时间小于当前时间，说明它之前被用过，需要重置。说白了就是一个用来清空的懒标记。

最后$dfs$的时候，不能用$p$数组遍历儿子，因为$dfs$到儿子的时候，$p$数组就变了，一个子树遍历完后回来，就会出现错误。

注意很坑的一点：点权可能为0，也就是说在树状数组里面可能出现死循环。给所有点的点权加一就行了。同时树状数组的上限也要加一。注意是在树状数组的修改和询问函数之内加，因为不能影响点本身的权值，计算答案要用的。

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) (x&(-x))
using namespace std;
const int maxn=5e4+10;
const int maxm=65538;
ll ans=0;
int Head[maxn],Next[maxn<<1],V[maxn<<1],p[maxn],cnt=0;
int siz[maxn],vis[maxn],mx[maxn],val[maxn],s[maxm],t[maxm];
int n,m,a,b,SIZ,tot,now,root=0;
const int RLEN=1<<18|1;
inline char nc() {
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob) ? -1 : *ib++;
}
inline int read() {
	char ch=nc(); int i=0,f=1;
	while(!isdigit(ch)) {if(ch=='-') f=-1; ch=nc();}
	while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
	return i*f;
}
inline int query(int x,int ret=0){
	for(int i=++x;i<=m;i+=lowbit(i)){
		if(t[i]<now) t[i]=now,s[i]=0;
		ret=max(ret,s[i]);
	}return ret;
}
inline void update(int x,int dis){
	for(int i=++x;i;i-=lowbit(i)){
		if(t[i]<now) t[i]=now,s[i]=0;
		s[i]=max(s[i],dis);
	}
}
inline void add(int u,int v){Next[++cnt]=Head[u],V[cnt]=v,Head[u]=cnt;}
inline void init(){now=cnt=root=0,mx[0]=SIZ=n;}
inline void getroot(int u,int f){
	siz[u]=1,mx[u]=0;
	for(int i=Head[u];i;i=Next[i]) if(!vis[V[i]]&& V[i]!=f)
		getroot(V[i],u),siz[u]+=siz[V[i]],mx[u]=max(mx[u],siz[V[i]]);
	mx[u]=max(mx[u],SIZ-mx[u]);if(mx[u]<mx[root]) root=u;
}
void getans(int u,int f,int M,int dep){
	ans=max(ans,1ll*M*(dep+query(M)+1));
	for(int i=Head[u];i;i=Next[i]) if(V[i]!=f && !vis[V[i]])
		getans(V[i],u,min(val[V[i]],M),dep+1);
}
void change(int u,int f,int M,int dep){
	update(M,dep);
	for(int i=Head[u];i;i=Next[i]) if(!vis[V[i]]&& V[i]!=f)
		change(V[i],u,min(val[V[i]],M),dep+1);
}
inline void dfs(int u){
	vis[u]=1,p[0]=0;
	for(int i=Head[u];i;i=Next[i]) if(!vis[V[i]]) p[++p[0]]=V[i];
	for(int i=1;i<=p[0];++i){
		getans(p[i],u,min(val[p[i]],val[u]),1);
		change(p[i],u,min(val[p[i]],val[u]),1);
	}++now;
	for(int i=p[0];i>=1;--i){
		getans(p[i],u,min(val[p[i]],val[u]),1);
		change(p[i],u,min(val[p[i]],val[u]),1);
	}++now;
	for(int i=Head[u];i;i=Next[i]) if(!vis[V[i]])
		SIZ=siz[V[i]],root=0,getroot(V[i],u),dfs(root);
}
int main(){
	n=read(),init();
	for(int i=1;i<=n;++i) val[i]=read(),m=max(m,val[i]+1);
	for(int i=1;i<n;++i) a=read(),b=read(),add(a,b),add(b,a);
	getroot(1,0),dfs(root);
	cout<<ans;
}