USACO section2.4 Cow Tours题解&代码

/*
ID:rainbow16
LANG:C++
TASK:cowtour
*/
#include<iostream>
#include<stdio.h>
#include<math.h>
using namespace std;
char temp[155];
int fa[1000],r[155][155],x[155],y[155],n;
double d[155][155],ma[155],ans,ansl=0xfffffff;
double dis(int i,int j)
{
return sqrt((double)((x[i]-x[j])*(x[i]-x[j]))+(double)((y[i]-y[j])*(y[i]-y[j])));
}
void init(void)
{
for(int i=1;i<1000;i++)
fa[i]=i;
}
int fin(int x)
{
if(fa[x]==x)
return x;
return fa[x]=fin(fa[x]);
}
void unio(int x,int y)
{
if(fin(x)!=fin(y))
fa[fin(x)]=fin(y);
}
void floyd(void)
{
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(i!=j && i!=k && j!=k && r[i][k] && r[j][k])
{
r[i][j]=2;
d[i][j]=d[i][j]?min(d[i][j],d[i][k]+d[j][k]):d[i][k]+d[j][k];
}

}
int main(void)
{
freopen("cowtour.in","r",stdin);
freopen("cowtour.out","w",stdout);
init();
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d%d",&x[i],&y[i]);
for(int i=1;i<=n;i++)
{
scanf("%s",&temp);
for(int j=1;j<=n;j++)
{
r[i][j]=temp[j-1]-'0';
if(r[i][j])
{
unio(i,j);
d[i][j]=d[j][i]=dis(i,j);
ma[i]=max(ma[i],d[i][j]);
ma[j]=max(ma[j],d[i][j]);
}
}
}
floyd();
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
if(r[i][j])
{
ma[i]=max(ma[i],d[i][j]);
ma[j]=max(ma[j],d[i][j]);
}
for(int i=1;i<=n;i++)
ans=max(ans,ma[i]);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(fin(i)!=fin(j))
ansl=min(ansl,ma[i]+ma[j]+dis(i,j));
printf("%f\n",ans>ansl?ans:ansl);
return 0;
}