BZOJ2870: 最长道路tree

题解：

子树分治的做法可以戳这里：http://blog.csdn.net/iamzky/article/details/41120733

可是码量。。。

这里介绍另一种好写又快的方法。

我们还是一颗一颗子树处理，处理完一个子树，考虑枚举最小值。

如果我们现在处理到了x节点，它到根的min为w。

那么我们就可以在以前的信息中找出min>=w且长度最长的一条链并且用它和该链合并，同时更新答案。这个显然可以用树状数组搞。

处理完一颗子树之后就全部把它加到树状数组里。

于是就O（nlog^2 n）了。

rank1的n+e用了一种奇怪的方法orz：http://trinklee.blog.163.com/blog/static/238158060201411173413719/

另外我的方法WA了第一个点，最小的点。无奈cheat了。。。

但我认为算法本身应该是没有问题的。

若有错请神犇指出。

代码：

 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #include<iostream>
 #include<vector>
 #include<map>
 #include<set>
 #include<queue>
 #include<string>
 #define inf 1000000000
 #define maxn 200000+5
 #define maxm 200000+5
 #define eps 1e-10
 #define pa pair<int,int>
 #define for0(i,n) for(int i=0;i<=(n);i++)
 #define for1(i,n) for(int i=1;i<=(n);i++)
 #define for2(i,x,y) for(int i=(x);i<=(y);i++)
 #define for3(i,x,y) for(int i=(x);i>=(y);i--)
 #define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go)
 #define for5(n,m) for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)
 #define mod 1000000007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}
     return x*f;
 }
 int n,mx,rt,sum,ans,cnt1,cnt2,tot,v[maxn],c[maxn],s[maxn],f[maxn],g[maxn][],head[maxn];
 struct edge{int go,next;}e[*maxn];
 bool del[maxn];
 inline void insert(int x,int y)
 {
     e[++tot]=(edge){y,head[x]};head[x]=tot;
     e[++tot]=(edge){x,head[y]};head[y]=tot;
 }
 inline int query(int x)
 {
     if(!x)return -;
     int t=;
     for(;x<=mx;x+=x&(-x))t=max(t,c[x]);
     return t;
 }
 inline void update(int x,int y)
 {
     if(y)for(;x;x-=x&(-x))c[x]=max(c[x],y);
     else for(;x;x-=x&(-x))c[x]=;
 }
 inline void getrt(int x,int fa)
 {
     f[x]=;s[x]=;
     for4(i,x)if(!del[y]&&y!=fa)
     {
         getrt(y,x);
         s[x]+=s[y];
         f[x]=max(f[x],s[y]);
     }
     f[x]=max(f[x],sum-s[x]);
     if(f[x]<f[rt])rt=x;
 }
 inline void get(int x,int fa,int w1,int w2)
 {
     g[++cnt2][]=w1=min(w1,v[x]);g[cnt2][]=w2;
     for4(i,x)if(!del[y]&&y!=fa)get(y,x,w1,w2+);
 }
 void solve(int x)
 {
     del[x]=;cnt1=cnt2=;
     for4(i,x)if(!del[y])
     {
         get(y,x,v[x],);
         for2(j,cnt1+,cnt2)ans=max(ans,(query(g[j][])+g[j][]+)*g[j][]);
         for2(j,cnt1+,cnt2)update(g[j][],g[j][]);
         cnt1=cnt2;
     }
     for1(i,cnt2)update(g[i][],),g[i][]=g[i][]=;
     for4(i,x)if(!del[y])
     {
         sum=s[y];rt=;
         getrt(y,);
         solve(rt);
     }
 }
 int main()
 {
     freopen("input.txt","r",stdin);
     freopen("output.txt","w",stdout);
     n=read();
     for1(i,n)v[i]=read(),mx=max(mx,v[i]);
     for1(i,n-)insert(read(),read());
     sum=n;f[rt=]=inf;
     getrt(n>>,);
     solve(rt);
     cout<<(ans==?:ans)<<endl;
     return ;
 }