hihoCoder-1097-Prim

这题就是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;
}

原文地址:https://www.cnblogs.com/xyqxyq/p/10366585.html