poj 2253 frogger 最短路

有N个点，青蛙A 在点1上，青蛙B在点2上，然后青蛙可以通过剩下的点来到达2点，求各个通路中最大的一段中的最小值~

题目连接http://poj.org/problem?id=2253

我的代码：

#include <stdio.h>
#include <string.h>
#include <math.h>
#include<algorithm>
#include <iostream>
using namespace std;

int n;

struct node
{
    int x,y;
} p[10000];
struct edge
{
    int u,v;
    double dis;
} e[100000];

int set[10000];
int cmp(struct edge a,struct edge b)
{
    return a.dis < b.dis;
}

int find(int x)
{
   if(set[x] != x)
   set[x] = find(set[x]);
   return set[x];
}
void merge(int a,int b)
{
    int fa,fb;
    fa = find(a);
    fb = find(b);
        set[fa] = fb;

    return ;
}
int main()
{
    int n;
    int cas;
    cas = 1;
    while(scanf("%d",&n)&&n)
    {
        int i,j,count;

        for(i = 0;i < n;i++)
        set[i] = i;
        for(i = 0; i < n; i++)
            scanf("%d %d",&p[i].x,&p[i].y);
        for(i = 0,count = 0; i < n; i++)
        {
            for(j = 0; j < i; j++)
            {
                double a,b;
                e[count].u = i;
                e[count].v = j;
                a = (p[i].x-p[j].x)*(p[i].x-p[j].x);
                b = (p[i].y-p[j].y)*(p[i].y-p[j].y);
                e[count++].dis = sqrt(a+b);
            }
        }
        double ans;
        ans = 0;
        sort(e,e+count,cmp);

        for(i = 0; i < count; i++)
        {

            if(find(0) == find(1))
            {
                ans = e[i].dis;
                break;
            }
            merge(e[i].u,e[i].v);
        }
        printf("Scenario #%d\nFrog Distance = %.3f\n\n",cas++,e[i-1].dis);
    }
    return 0;
}