prim算法之 SDUT2937 人活着系列之寻找最完美的人生

Kaka_chake

于 2018-08-20 16:25:29 发布

阅读量226

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/Kaka_chake/article/details/81873345

本文介绍了一个使用C语言实现的Prim算法，该算法用于在一个带权重的无向图中寻找最小生成树。通过初始化距离数组和使用邻接矩阵来记录顶点之间的连接情况，程序能够有效地找出连接所有顶点且权重最小的边集合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <stdio.h>
#include <stdlib.h>
#include <cstring>
int m[1005][1005], book[1005], dist[1005], sum;

void prim(int n)
{
    int i, j;
    for( i = 1; i <= n; i++)
    {
        dist[i] = m[i][1];
    }
    book[1] = 1;
    for(i = 2 ; i <= n; i++)
    {
       int  pos = 0;
       int Min = 100001;
        for( j = 1; j <= n ;j++)
        {
            if(!book[j]&&Min>dist[j])
            {
                Min = dist[j];
                pos = j;
            }
        }
        sum += dist[pos];
        book[pos] = 1;
        for(j = 1 ; j <= n ; j++)
        {
            if(!book[j]&&dist[j]>m[pos][j])
            {
                dist[j] = m[pos][j];
            }
        }
    }
    printf("%d\n", sum);
}

int main()
{
    int t, n , i, j ,k, a[1005];
    scanf("%d", &t);
    while(t--)
    {
        sum = 0;
        memset(book, 0, sizeof(book));
        scanf("%d", &n);
        for( i = 1 ; i <= n ;i++)
        {
            scanf("%d", &a[i]);
        }
        for( i = 1 ; i <= n ;i++)
        {
            for( j = 1 ; j <= n; j++)
            {
                m[i][j] = m[j][i] = 3*(a[i]+a[j]) + 7;
            }
        }
        prim(n);
    }
    return 0;
}