这题就是prim的板子题,不过如果用end每次初始化为-1的话,我们就不需要对于每次选中的下一个点进行判断是否选中了,因为每次外层循环第一次进入都是可以的。
然后还很 (i=1;i<=n;i++) 正好对应上,因为n个点,n-1条边,跑n-1次就够了,但是我们第一次是让原点进入更新d[ ]数组值的。
其实prim和dijkstra还是很像的。
#include <cstdio>
#include <cstring>
const int INF = 0x3f3f3f3f;
int map[1010][1010], d[1010];
int vis[1010];
int main()
{
int n;
scanf("%d", &n);
for (int i = 1; i <= n;i++) {
for (int j = 1; j <= n;j++) {
scanf("%d",&map[i][j]);
}
}
memset(vis, 0, sizeof(vis));
memset(d, INF, sizeof(d));
d[1]=0;
int end, res=0;
for (int i = 1; i <= n;i++) {
end = -1;//从第一个点进入,更新了d[],并且保证了不会有空点
for (int j = 1; j <= n;j++) {
if (!vis[j]&&(end==-1||d[end]>d[j])) {
end = j;
}
}
vis[end] = 1;
res += d[end];
for (int j = 1; j <= n;j++) {
if (!vis[j]&&d[j]>map[end][j]) {
d[j] = map[end][j];
}
}
}
printf("%d
", res);
return 0;
}