ZOJ - 1586QS Network prim

最新推荐文章于 2022-04-11 13:16:53 发布

原创最新推荐文章于 2022-04-11 13:16:53 发布 · 230 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#prim #最小生成树

最小生成树专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种基于Prim算法求解最小生成树问题的方法，并通过C++代码实现了该算法。算法首先初始化所有顶点到无穷大成本，然后选择一个起始顶点开始构建最小生成树，每次迭代都寻找当前已包含顶点到未包含顶点中成本最低的边进行添加。

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

建图边值为电缆得钱数加上两边电话的钱数，贴模板

#include <stdio.h>
#include <string.h>
#include <iostream>
#include<algorithm>
#include <vector>
#include <queue>
#include <string>
#include <math.h>
#include <stdlib.h>
using namespace std;
#define INF 0x3f3f3f3f
#define mem(arr,a) memset(arr,a,sizeof(arr))
#define V 1000+5
#define LL long long int
#define E 320000
#define pow(a) ((a)*(a))
int n, m, t;
int cost[V][V];
int vis[V];
int minCost[V];
int arr[V];
int sum;
void prim(){
    mem(minCost, INF);
    minCost[1] = 0;
    mem(vis, 0);
    while (1){
        int v = -1;
        for (int i = 1; i <= n; i++){
            if (!vis[i] && (v == -1 || minCost[i] < minCost[v]))v = i;
        }
        if (v == -1)break;
        vis[v] = 1;
        sum += minCost[v];
        for (int i = 1; i <= n; i++){
            if (minCost[i]>cost[v][i]){
                minCost[i] = cost[v][i];

            }
        }
    }
    cout << sum << endl;
}
int main(){
    cin >> t;
    while (t--){
        cin >> n;
        sum = 0;
        for (int i = 1; i <= n; i++)cin >> arr[i];
        for (int i = 1; i <= n; i++){
            for (int j = 1; j <= n; j++){
                cin >> cost[i][j];
                if (i!=j)
                cost[i][j] += arr[i] + arr[j];
            }
        }
        prim();
    }
}