49 lines
1.3 KiB
C++
49 lines
1.3 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
const int N = 310;
|
|||
|
|
|
|||
|
|
int n;
|
|||
|
|
int g[N][N];
|
|||
|
|
int dis[N];
|
|||
|
|
bool st[N];
|
|||
|
|
int res; // 最小生成树里面边的长度之和
|
|||
|
|
|
|||
|
|
void prim() {
|
|||
|
|
memset(dis, 0x3f, sizeof dis); // 初始化所有距离为INF
|
|||
|
|
dis[0] = 0; // 超级源点是在生成树中的
|
|||
|
|
|
|||
|
|
for (int i = 0; i <= n; i++) { // 注意：这里因为引入了超级源点，所以点的个数是n+1
|
|||
|
|
int t = -1;
|
|||
|
|
for (int j = 0; j <= n; j++)
|
|||
|
|
if (!st[j] && (t == -1 || dis[t] > dis[j])) t = j;
|
|||
|
|
|
|||
|
|
if (i) res += dis[t];
|
|||
|
|
// 有超级源点的题，是必然存在最小生成树的
|
|||
|
|
// 注意这里也是需要从0~n共n+1个
|
|||
|
|
for (int j = 0; j <= n; j++)
|
|||
|
|
if (!st[j] && dis[j] > g[t][j])
|
|||
|
|
dis[j] = g[t][j];
|
|||
|
|
st[t] = true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int main() {
|
|||
|
|
cin >> n;
|
|||
|
|
// 建立超级源点(0 <-> 1~n )，点权转化为超级源点到此节点的边权
|
|||
|
|
for (int i = 1; i <= n; i++) {
|
|||
|
|
int c;
|
|||
|
|
cin >> c;
|
|||
|
|
g[i][0] = g[0][i] = c;
|
|||
|
|
}
|
|||
|
|
// 本题是按矩阵读入的，不是按a,b,c方式读入的
|
|||
|
|
for (int i = 1; i <= n; i++)
|
|||
|
|
for (int j = 1; j <= n; j++)
|
|||
|
|
cin >> g[i][j];
|
|||
|
|
|
|||
|
|
// 利用prim计算最小生成树
|
|||
|
|
prim();
|
|||
|
|
cout << res << endl;
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|