【洛谷】P1522 [USACO2.4]牛的旅行 Cow Tours

#include <bits/stdc++.h>
using namespace std;

#define maxn 160
#define inf  1e20
struct node{
    int x,y;
};
double calS(const node &n1,const node &n2){
    return sqrt((n1.x-n2.x)*(n1.x-n2.x)+(n1.y-n2.y)*(n1.y-n2.y));
}
int nv;
int G[maxn][maxn];
node nnode[maxn];

int field[maxn];
double dist[maxn][maxn];//dist[i][j] -> i,j间的最短距离

void dfs(int root,int id){
    field[root]=id;
    for(int j=0;j<nv;j++){
        if(dist[root][j]<inf && !field[j]) dfs(j,id);
    }
}


void floyd(){
    for(int k=0;k<nv;k++){
        for(int i=0;i<nv;i++){
            for(int j=0;j<nv;j++){
                if(dist[i][k]<inf && dist[k][j]<inf && dist[i][k]+dist[k][j]<dist[i][j]) 
                    dist[i][j]=dist[i][k]+dist[k][j];
            }
        }
    }
}

int main(){
    scanf("%d",&nv);

    int t1,t2;
    for(int i=0;i<nv;i++){
        scanf("%d %d",&t1,&t2);
        nnode[i].x=t1,nnode[i].y=t2;
    }

    for(int i=0;i<nv;i++){
        for(int j=0;j<nv;j++){
            scanf("%1d",&G[i][j]);
        }
    }

    //dist[][]初始化:
    for(int i=0;i<nv;i++){
        for(int j=0;j<nv;j++){
            //直接相连的最短距离=坐标计算的距离
            if(G[i][j] || i==j) dist[i][j]=calS(nnode[i],nnode[j]);
            else                dist[i][j]=inf;
        }
    }
    
    //1:dfs访问每个连通分量
    int id=0;
    for(int i=0;i<nv;i++){
        if(!field[i]) dfs(i,++id);//id必须至少从1开始,不然field[]没法标记访问过
    }

    //2:floyd求最短路径
    floyd();

    //3:每个牧场的直径
    double maxS[id+1];//注意id从1开始
    double toNodeMaxS[maxn];
    for(int i=0;i<nv;i++){
        int id=field[i];
        for(int j=0;j<nv;j++){//怕影响toNodeMaxS[]的值,所以j没从i+1开始取
            if(dist[i][j]<inf)
                toNodeMaxS[i]=max(dist[i][j],toNodeMaxS[i]);
        }
        maxS[id]=max(toNodeMaxS[i],maxS[id]);//每个牧场的直径
    }

    //4:新牧场直径:连接后的最短路端点(两牧区)可能会改变
    //由于新边引起的改变,只会影响与两端点的之前就连通的点彼此之间的距离
    //直径=到i点的最长距离+ij距离+到j点的最长距离
    double maxD=-1,minD=inf;
    for(int i=0;i<nv;i++){
        for(int j=i+1;j<nv;j++){//无向图,连接i->j和j->i一样
            if(field[i]!=field[j]){//两牧场
                maxD=max(toNodeMaxS[i]+calS(nnode[i],nnode[j])+toNodeMaxS[j],max(maxS[field[i]],maxS[field[j]]));//最大直径
                minD=min(maxD,minD);
            }
        }
    }
    
    printf("%.6f",minD);

    return 0;
}