bzoj 2870 最长道路tree——边分治

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=2870

关于边分治：https://www.cnblogs.com/Khada-Jhin/p/10154994.html

自己写了那种把一个孩子连向自己，其他孩子连新建节点的重构图方法。

如果 vector 里最后一个元素恰好是父亲的话，末尾就会有一个新建节点只有一个孩子。

solve( ) 完之后要继续 get_rt( ) ，想知道两边的点数 ts ；可以发现 to[ cr ] 的部分点数恰好是 siz[ to[ cr ] ] ，另一部分就是 s - siz[ to[ cr ] ] 了。

要递归的时候，如果 ts == 1 ，就不递归了。

关于这道题：https://www.cnblogs.com/Miracevin/p/10430192.html

虚点的点权就是它建出它的那个实点的点权；路径上的点数等于路径上的实边条数 + 1 。

找出两边的路径，分别 sort 然后双指针即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define pb push_back
#define ll long long
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
ll Mx(ll a,ll b){return a>b?a:b;}
ll Mn(ll a,ll b){return a<b?a:b;}
const int N=5e4+,M=N<<,INF=7e4;
int n,vl[M],hd[M],xnt=,to[M<<],nxt[M<<],w[M<<];//xnt=1
int siz[M],mn,Rt,tot[]; bool vis[M]; ll ans;
vector<int> vt[N];
struct Node{
  int dis,mn;
  Node(int d=,int m=):dis(d),mn(m) {}
  bool operator< (const Node &b)const
  {return mn<b.mn;}
}a[][M];
void add(int x,int  y,int z)
{
  to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;w[xnt]=z;
  to[++xnt]=x;nxt[xnt]=hd[y];hd[y]=xnt;w[xnt]=z;
}
void Rbuild(int cr,int fa)
{
  for(int i=,lm=vt[cr].size()-,v,lst=;i<=lm;i++)
    {
      if((v=vt[cr][i])==fa)continue;
      if(!lst) add(cr,v,), lst=cr;
      else if(i==lm) add(lst,v,);
      else
    {
      n++; vl[n]=vl[cr];
      add(lst,n,); add(n,v,); lst=n;
    }
    }
  for(int i=,lm=vt[cr].size(),v;i<lm;i++)
    if((v=vt[cr][i])!=fa)Rbuild(v,cr);
}
void get_rt(int cr,int fa,int s)
{
  siz[cr]=;
  for(int i=hd[cr],v,d;i;i=nxt[i])
    if(!vis[i>>]&&(v=to[i])!=fa)
      {
    get_rt(v,cr,s); siz[cr]+=siz[v];
    d=Mx(siz[v],s-siz[v]);
    if(d<mn)mn=d, Rt=i;
      }
}
void dfs(int cr,int fa,int lj,int mn,bool fx)
{
  mn=Mn(mn,vl[cr]); a[fx][++tot[fx]]=Node(lj,mn);
  for(int i=hd[cr],v;i;i=nxt[i])
    if(!vis[i>>]&&(v=to[i])!=fa)
      dfs(v,cr,lj+w[i],mn,fx);
}
void cz()
{
  for(int i=;i<=;i++)
    sort(a[i]+,a[i]+tot[i]+);
  int p0=tot[]+, lj=, tw=w[Rt]+;//+1
  for(int i=tot[];i;i--)
    {
      int tmn=a[][i].mn;
      while(p0>&&a[][p0-].mn>=tmn)
    p0--, lj=Mx(lj,a[][p0].dis);
      if(p0>tot[])continue;//
      ans=Mx(ans,(ll)tmn*(lj+a[][i].dis+tw));
    }
  p0=tot[]+; lj=;
  for(int i=tot[];i;i--)
    {
      int tmn=a[][i].mn;
      while(p0>&&a[][p0-].mn>=tmn)
    p0--, lj=Mx(lj,a[][p0].dis);
      if(p0>tot[])continue;//
      ans=Mx(ans,(ll)tmn*(lj+a[][i].dis+tw));
    }
}
void solve(int cr,int s)
{
  vis[cr>>]=;
  tot[]=tot[]=;
  dfs(to[cr],,,INF,); dfs(to[cr^],,,INF,);
  cz();
  int v=to[cr], ts=siz[v];
  if(ts>){mn=N; get_rt(v,,ts); solve(Rt,ts);}
  ts=s-ts; v=to[cr^];// s-ts not s-siz[v] for siz[v] may changed!!!
  if(ts>){mn=N; get_rt(v,,ts); solve(Rt,ts);}
}
int main()
{
  n=rdn();
  for(int i=;i<=n;i++)vl[i]=rdn();
  for(int i=,u,v;i<n;i++)
    {
      u=rdn();v=rdn(); vt[u].pb(v); vt[v].pb(u);
    }
  Rbuild(,);
  mn=N;get_rt(,,n);solve(Rt,n);
  printf("%lld\n",ans);
  return ;
}