Constructing Minimum Roads（修建最短距离公路）问题

简而言之，就是N个村庄互相连通，在指定个别已修好的路的前提下求最短公路距离。

结合数据结构的普里姆算法和克鲁斯卡尔算法求出。

算法输入第一行为村庄个数，然后各行就是个邻接矩阵，最后指出有几条已修的公路，并分别用编号列出；算法输出即为所有需要修建的各段公路长度。

 

 

#include <iostream>
#include <algorithm>//<algorithm>是c++特有的STL模板的算法头文件，包含了一些特定的算法函数，此处用的是sort()函数进行排序
using namespace std;
#define N 100

struct node
{
	int u,v,w;//u为源村庄编号，v为目的村庄编号，w为二者之间的距离
	
}edge[N*N];
int parent[N];
bool cmp(node a,node b)//作为sort()的第三个比较参数来确定为距离升序排列
{
    return a.w<=b.w;
    
}
int find(int a)
{
    if(a!=parent[a])
        return find(parent[a]);
    else return a;
}
int kruskal(int n,int m)//克鲁斯卡尔算法
{
    sort(edge,edge+m,cmp);//对edge[0~m]内的数进行升序排列
    int i,x,y,ans=0;
    for(i=0;i<m;i++)
    {
        x=edge[i].u;
        y=edge[i].v;
        x=find(x);
        y=find(y);
        if(x!=y)
        {
            ans+=edge[i].w;
            parent[y]=x;
        }
    }
    return ans;
}
int main()
{
    int n,q,k,i,j,m;
    while(1)
    {  
		cin>>n;
		if(n<=0) break;//当输入的村庄数目为非正时结束循环
        m=0;
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++)
            {    
                cin>>k;//输入村庄之间的距离
                if(i>=j) continue;//按下三角统计两两村庄之间的距离
                edge[m].u=i;
                edge[m].v=j;
                edge[m].w=k;
                m++;
            }
        }
        for(k=1;k<=n;k++) parent[k]=k;
        cin>>q;//输入已修好几条路
        for(k=1;k<=q;k++)
        {
            cin>>i>>j;//i、j为已修好路的村庄编号
            i=find(i);
            j=find(j);
            parent[j]=i;
        }
        cout<<kruskal(n,m)<<endl;
    }
    return 0;
}

 

假设有三个村庄A、B、C，分别用编号1，2，3代表，其中AB相距5km，AC相距15km，BC相距10km，初始状态为BC之间的公路已经修好，则效果图如下：