c++题目_P1546 [USACO3.1] 最短网络 Agri-Net

题目背景

Farmer John 被选为他们镇的镇长!他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。当然,他需要你的帮助。

题目描述

FJ 已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。为了用最小的消费,他想铺设最短的光纤去连接所有的农场。

你将得到一份各农场之间连接费用的列表,你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过 105105。

输入格式

第一行农场的个数 𝑁N(3≤𝑁≤1003≤N≤100)。

接下来是一个 𝑁×𝑁的矩阵,表示每个农场之间的距离。理论上,他们是 𝑁 行,每行由 𝑁 个用空格分隔的数组成,实际上,由于每行 8080 个字符的限制,因此,某些行会紧接着另一些行。当然,对角线将会是 00,因为不会有线路从第 𝑖个农场到它本身。

输出格式

只有一个输出,其中包含连接到每个农场的光纤的最小长度。

输入输出样例

输入 #1复制

复制代码
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

输出 #1复制

复制代码
28

说明/提示

题目翻译来自NOCOW。

USACO Training Section 3.1

这一题我所提供的代码示例基于 Prim 算法的最小生成树解法。

cpp 复制代码
#include <iostream>
#include <vector>
#include <climits>
using namespace std;

int findMinDist(vector<int>& dist, vector<bool>& included, int N) {
    int minDist = INT_MAX;
    int minDistIdx = -1;

    for (int i = 0; i < N; i++) {
        if (!included[i] && dist[i] < minDist) {
            minDist = dist[i];
            minDistIdx = i;
        }
    }

    return minDistIdx;
}

int calculateMinLength(vector<vector<int>>& graph, int N) {
    vector<int> dist(N, INT_MAX);
    vector<bool> included(N, false);

    dist[0] = 0;

    for (int i = 0; i < N - 1; i++) {
        int u = findMinDist(dist, included, N);
        included[u] = true;

        for (int v = 0; v < N; v++) {
            if (graph[u][v] && !included[v] && graph[u][v] < dist[v]) {
                dist[v] = graph[u][v];
            }
        }
    }

    int minLength = 0;
    for (int i = 0; i < N; i++) {
        minLength += dist[i];
    }

    return minLength;
}

int main() {
    int N;
    cin >> N;

    vector<vector<int>> graph(N, vector<int>(N));

    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            cin >> graph[i][j];
        }
    }

    int minLength = calculateMinLength(graph, N);
    cout << minLength << endl;

    return 0;
}
相关推荐
JohnFF7 分钟前
48. 旋转图像
数据结构·算法·leetcode
bbc1212267 分钟前
AT_abc306_b [ABC306B] Base 2
算法
满怀101510 分钟前
Python入门(5):异常处理
开发语言·python
張萠飛13 分钟前
Linux的TCP连接数到达2万,其中tcp_tw、tcp_alloc、tcp_inuse都很高,可能出现什么问题
linux·网络·tcp/ip
攀小黑13 分钟前
Java 多线程加锁 synchronized 关键字 字符串当做key
java·开发语言
生锈的键盘15 分钟前
推荐算法实践:movielens数据集
算法
董董灿是个攻城狮16 分钟前
Transformer 通关秘籍9:词向量的数值实际上是特征
算法
每次的天空23 分钟前
Kotlin 作用域函数:apply、let、run、with、also
android·开发语言·kotlin
代码AC不AC25 分钟前
【数据结构】队列
c语言·数据结构·学习·队列·深度讲解
林泽毅25 分钟前
SwanLab x EasyR1:多模态LLM强化学习后训练组合拳,让模型进化更高效
算法·llm·强化学习