SPOJ TWOPATHS Two Paths

#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int INF=0x7fffffff;
const int maxn=100002;
struct Edge
{
    int to,pre;
    Edge (int to=0,int pre=0):to(to),pre(pre){}
};
Edge edge[maxn*2];
int head[maxn],pos;
bool vis[maxn];
bool in_line[maxn];
int ko[maxn],pos_ko;
int dp[maxn][2];
int dp1[maxn],dp2[maxn];
struct info
{
    int p,pre;
};
info  que[maxn];
void inint()
{
    memset(head,-1,sizeof(head));
    pos=pos_ko=0;
    memset(dp,0,sizeof(dp));
    memset(in_line,false,sizeof(in_line));
    memset(dp1,0,sizeof(dp1));
    memset(dp2,0,sizeof(dp2));
}
void add_edge(int s,int to)
{
    edge[pos]=Edge(to,head[s]);
    head[s]=pos++;
}
void make(int P)
{
    if(que[P].pre!=-1)make(que[P].pre);
    ko[pos_ko++]=que[P].p;
    in_line[que[P].p]=true;
}
int bfs(int t,bool flag)
{
    memset(vis,false,sizeof(vis));
    int h=0,r=1;
    que[0].p=t;
    que[0].pre=-1;
    vis[t]=true;
    int x;
    while(h<r)
    {
        x=que[h++].p;
        for(int i=head[x];~i;i=edge[i].pre)
        {
            Edge &tmp=edge[i];
            if(!vis[tmp.to])
                que[r].p=tmp.to,
                que[r++].pre=h-1,
                vis[tmp.to]=true;
        }
    }
    if(flag)make(r-1);
    return que[r-1].p;
}
void get_it(int key,int t)
{
    if(key>dp[t][0])dp[t][1]=dp[t][0],
                    dp[t][0]=key;
    else if(key>dp[t][1])
                    dp[t][1]=key;
}
int Max;
int dfs(int pa,int &s,int &t)
{
    int key,ans=0;
    for(int w=head[s];~w;w=edge[w].pre)
    {
        Edge &tmp=edge[w];
        if(tmp.to==pa||in_line[tmp.to])continue;
        key=dfs(s,tmp.to,t);
        if(pa==-1) get_it(key,t);
        if(pa!=-1)
        {
            if(ans)
                Max=max(Max,ans-1+key);
            else
                Max=max(Max,key);
        }
        ans=max(ans,key+1);
    }
    if(pa==-1)Max=max(Max,ans-2);
    return ans==0?1:ans;
}
void solve(int &n)
{

    long long  ans=0;
    Max=0;
    int p1=bfs(1,false),p2=bfs(p1,true);
    int Max1;
    for(int i=0;i<pos_ko;i++)
        dfs(-1,ko[i],i);
    ans=(pos_ko-1)*Max;
    --pos_ko;
    for(int i=1;i<pos_ko;i++)
    {
        dp1[i]=max(dp1[i-1],i+dp[i][0]);
        dp1[i]=max(dp1[i],dp[i][0]+dp[i][1]);
    }
    for(int i=pos_ko-1;i>=0;i--)
    {
        dp2[i]=max(dp2[i+1],pos_ko-i+dp[i][0]);
        dp2[i]=max(dp2[i],dp[i][0]+dp[i][1]);
    }
    for(int i=1;i<pos_ko;i++)
        ans=max(ans,(long long)dp1[i]*dp2[i+1]);
    ans=max(ans,(long long )0);
    printf("%lld\n",ans);
}
int main()
{
    int n;
    int a,b;
    while(~scanf("%d",&n))
    {
        inint();
        for(int i=2;i<=n;i++)
        {
            scanf("%d%d",&a,&b);
            add_edge(a,b);
            add_edge(b,a);
        }
        solve(n);
    }
    return 0;
}