Description
农民约翰被选为他们镇的镇长!他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。当然,他需要你的帮助。约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。为了用最小的消费,他想铺设最短的光纤去连接所有的农场。你将得到一份各农场之间连接费用的列表,你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000
Input
第一行: 农场的个数,N(3<=N<=100)。
第二行…结尾: 后来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个农场到它本身。
Output
只有一个输出,其中包含连接到每个农场的光纤的最小长度。
Sample Input
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
Sample Output
28
思路
该题和这一道有什么差别吗??
因为发现读入的是邻接矩阵,所以我们使用prim+邻接矩阵读入,不为什么,主要是比较快,好理解
code:
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
struct f{
int x,y;
};
int m[101];
bool operator <(const f &a,const f &b)
{
return a.x>b.x;
}
priority_queue<f> p;
int a[101][101];
int main()
{
m[1]=-0x7f7f7f7f+1;
int n;
cin>>n;
for (int i=1;i<=n;i++)
{
for (int j=1;j<=n;j++)
{
cin>>a[i][j];
}
}
int n2=n-1;
long long s=0;
f o,ow;
for (int i=2;i<=n;i++)
{
o.x=a[1][i],o.y=i;
m[i]=o.x;
p.push(o);
}
while (n2!=0)
{
n2--;
while (p.size()&&p.top().x>m[p.top().y]) p.pop();
o=p.top();
p.pop();
s+=o.x;
for (int i=1;i<=n;i++)
{
if (a[o.y][i]<m[i])
{
m[i]=a[o.y][i];
ow.x=m[i],ow.y=i;
p.push(ow);
}
}
m[o.y]=-0x7f7f7f7f+1;
}
cout<<s;
return 0;
}