21.prim算法 最优布线问题(wire.cpp)

prim算法 最优布线问题(wire.cpp)

【例4-10】、最优布线问题(wire.cpp)

【问题描述】

　　学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。

当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。

　　现在由你负责连接这些计算机，任务是使任意两台计算机都连通（不管是直接的或间接的）。

【输入格式】

　　输入文件wire.in，第一行为整数n（2<=n<=100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

【输出格式】

　　输出文件wire.out，一个整数，表示最小的连接费用。

【输入样例】

　　3

　　0 1 2

　　1 0 1

　　2 1 0

【输出样例】

2 （注：表示连接1和2，2和3，费用为2）

代码：

#include

using namespace std;

#include

#include

#include

#define maxn 101

int dis[maxn][maxn],minn[maxn],visit[maxn]={0},n;//0 shi mei jin ru 1 shi jin ru de le

void input();

void prim();

long long tot=0;

int main()

{

input();

prim();

printf("%I64d",tot);

return 0;

}

void input()

{

scanf("%d",&n);

for(int i=1;i<=n;++i)

for(int j=1;j<=n;++j)

scanf("%d",&dis[i][j]);

}

void prim()

{

memset(minn,0x7f,sizeof(minn));//把每个蓝点到树（白点）的距离都设成很大

minn[1]=0;//从一开始成树

for(int i=1;i<=n;++i)//因为要把n个点都进入最小生成树，那么就循环n次，prim循环的是点的数目

{

int k=0;

for(int j=1;j<=n;++j)

if(!visit[j]&&minn[j]是个大数

k=j;//找出最小值

visit[k]=1;//始终把最小的边放入最小生成树

for(int l=1;l<=n;++l)

if(!visit[l]&&dis[k][l]之所以有！visit的条件是因为，我们只是把蓝点到白点的距离更新，不更新树中的

minn[l]=dis[k][l];//

}

for(int i=1;i<=n;++i)

tot+=minn[i];//统计每个点到最小生成树的权值，加起来就行了

}