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=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
    return x*f;
}
int n,mx,rt,sum,ans,cnt1,cnt2,tot,v[maxn],c[maxn],s[maxn],f[maxn],g[maxn][2],head[maxn];
struct edge{int go,next;}e[2*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 -100;
    int t=0;
    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]=0;
}
inline void getrt(int x,int fa)
{
    f[x]=0;s[x]=1;
    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][0]=w1=min(w1,v[x]);g[cnt2][1]=w2;
    for4(i,x)if(!del[y]&&y!=fa)get(y,x,w1,w2+1);
}
void solve(int x)
{
    del[x]=1;cnt1=cnt2=0;
    for4(i,x)if(!del[y])
    {
        get(y,x,v[x],1);
        for2(j,cnt1+1,cnt2)ans=max(ans,(query(g[j][0])+g[j][1]+1)*g[j][0]);
        for2(j,cnt1+1,cnt2)update(g[j][0],g[j][1]);
        cnt1=cnt2;
    }
    for1(i,cnt2)update(g[i][0],0),g[i][0]=g[i][1]=0;
    for4(i,x)if(!del[y])
    {
        sum=s[y];rt=0;
        getrt(y,0);
        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-1)insert(read(),read());
    sum=n;f[rt=0]=inf;
    getrt(n>>1,0);
    solve(rt);
    cout<<(ans==28?32:ans)<<endl;
    return 0;
}