poj2485——Highways

题目大意：在N个村庄之间建高铁，让所有村庄都相互可达，问如何建高铁能让所有高铁路段中的最长段最小

输入：（可以有很多case）

            case个数T（然后空一行）

            N(3 <= N <= 500)

            第i个村庄到第j个村庄的距离（共N行，每行N个数，空格隔开，相当于一个距离矩阵）

            （每个case后空一行）距离范围 [1, 65536]

输出：所有高铁路段中的最长段

分析：求最小生成树的最长段。prim和krustal都可以。

           prim算法：设点集V和边集E，初始V中放入图中任意一点，在其余点中找出距离V中点最近的点加入V，并将这条边加入E，以此类推，直到V中包含所有点。

           krustal算法：设最小生成树边集E，在图中所有边中找出最短边，如果和E中的边不成环，就加入E，直到E中有n-1条边。

代码：

方法一：prim算法。

#include<cstdio>
#include<cstring>
using namespace std;
int ma[520][520];
int dist[520];
int vis[520];
int n;
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        memset(ma,sizeof(ma),0x3f3f3f3f);
        for(int i = 0;i < n;i++){
            for(int j = 0;j < n;j++){
                scanf("%d",&ma[i][j]);
            }
        }
        for(int i = 0;i < n;i++){
            dist[i] = ma[0][i];
            vis[i] = 0;
        }
        vis[0] = 1;
        int maxLen = 0;
        for(int i = 0;i < n - 1;i++){
            int u = 0,minx = 0x3f3f3f3f;
            for(int j = 0;j < n;j++){
                if(!vis[j] && dist[j] < 0x3f3f3f3f && dist[j] < minx){
                    u = j;
                    minx = dist[j];
                }
            }
            vis[u] = 1;
            if(minx > maxLen) maxLen = minx;
            for(int j = 0;j < n;j++){
                if(!vis[j] && dist[j] > ma[u][j]){
                    dist[j] = ma[u][j];
                }
            }
        }
        printf("%d\n",maxLen);
    }
}

方法二：krustal算法。转载自https://www.cnblogs.com/shenben/p/5630447.html

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
#define N 510
struct node{
    int x,y,v;
}e[N*N];
int t,fa[N];
bool cmp(const node &a,const node &b){
    return a.v<b.v;
}
int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int main(){
    scanf("%d",&t);
    while(t--){
        int n,cnt=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++) fa[i]=i; //并查集初始化根节点为自己本身
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                int x;
                scanf("%d",&x);
                if(x){
                    e[++cnt].x=i;e[cnt].y=j;e[cnt].v=x;
                }
            }
        }
        int k=0,res=0;
        sort(e+1,e+cnt+1,cmp);
        for(int i=1;i<=cnt;i++){ //按从小到大的顺序遍历每条边
            int fx=find(e[i].x),fy=find(e[i].y);
            if(fx!=fy){   //加入这条边不成环，则将这条边加入最小生成树边集E
                fa[fy]=fx;
                k++;
            }
            if(k==n-1){
                res=e[i].v;//因为排完序了，所以最后一个就是最大的
                break;
            }
        }
        printf("%d\n",res);
        for(int i=1;i<=n;i++) fa[i]=0;//养成清零好习惯 
        for(int i=1;i<=n;i++) e[i]=e[0];
    }
    return 0;
}