【题目描述】
学校有nn台计算机,为了方便数据传输,现要将它们用数据线连接起来。两台计算机被连接是指它们有数据线连接。由于计算机所处的位置不同,因此不同的两台计算机的连接费用往往是不同的。
当然,如果将任意两台计算机都用数据线连接,费用将是相当庞大的。为了节省费用,我们采用数据的间接传输手段,即一台计算机可以间接的通过若干台计算机(作为中转)来实现与另一台计算机的连接。
现在由你负责连接这些计算机,任务是使任意两台计算机都连通(不管是直接的或间接的)。
【输入】
第一行为整数n(2≤n≤1002),表示计算机的数目。此后的n行,每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。
【输出】
一个整数,表示最小的连接费用。
【输入样例】
3 0 1 2 1 0 1 2 1 0
【输出样例】
2
【提示】
注:表示连接1和2,2和3,费用为2。
#include <bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
int g[101][101];
int n,ans=0;
struct Close {
int adj; //邻接点
int cost; //代价
};
Close c[101];
int getMin()
{
int min=INF,pos=0;
for(int i=1;i<=n;i++)
{
if(c[i].cost!=0&&c[i].cost<min)
{
min=c[i].cost;
pos=i;
}
}
return pos;
}
void Prim(int u)
{
for(int i=1;i<=n;i++)
{
if(i!=u)
{
c[i].adj=u;
c[i].cost=g[u][i]; //存放连着U的顶点以及最小路径
}
}
c[u].cost=0; //初始U={u}
for(int i=1;i<=n;i++)
{
int pos=getMin();
if(c[pos].adj!=0)
{
ans+=g[pos][c[pos].adj];
}
c[pos].cost=0;
for(int j=1;j<=n;j++)
{
if(c[j].cost!=0&&g[pos][j]<c[j].cost)
{
c[j].cost=g[pos][j];
c[j].adj=pos;
}
}
}
}
int main()
{
memset(g,0x3f,sizeof(g));
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>g[i][j];
Prim(1);
cout<<ans;
return 0;
}