题目大意:
学校有n台计算机,为了方便数据传输,现要将它们用数据线连接起来。两台计算机被连接是指它们之间有数据线连接。由于计算机所处的位置不同,因此不同的两台计算机的连接费用往往是不同的。
当然,如果将任意两台计算机都用数据线连接,费用将是相当庞大的。为了节省费用,我们采用数据的间接传输手段,即一台计算机可以间接的通过若干台计算机(作为中转)来实现与另一台计算机的连接。
现在由你负责连接这些计算机,你的任务是使任意两台计算机都连通(不管是直接的或间接的)。
思路:
最!
小!
生!
成!
树!
这道题用O(n^3)的方法会爆,5000的数据!,所以只能用O(n^2)的方法。
代码:
#include <cstdio>
using namespace std;
int a[5001][5001],n,b[5001],k,sum,minn,c[5001];
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
for (int j=1;j<=n;j++)
{
scanf("%d",&a[i][j]);
}
b[i]=i;
}
for (int i=1;i<=n;i++)
if (a[1][i]!=0) c[i]=a[1][i]; //初始化,判断第一台电脑到其他电脑的最近距离
for (int q=1;q<=n-1;q++)
{
minn=2147483647;
for (int i=1;i<=n;i++)
{
if (c[i]<minn&&c[i]!=0) //求距离集合最近的一个在集合外的电脑
{
minn=c[i];
k=i; //记录
}
}
c[k]=0;
b[k]=1; //进入集合
sum+=minn;
for (int i=1;i<=n;i++)
if (b[i]!=1&&c[i]>a[k][i]) c[i]=a[k][i]; //重新判断集合到每个电脑的最近距离
}
printf("%d\n",sum);
return 0;
}
(这道题与最短网络极其相似)