试题编号: | 201412-4 |
试题名称: | 最优灌溉 |
时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: |
问题描述
雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。
为了灌溉,雷雷需要建立一些水渠,以连接水井和麦田,雷雷也可以利用部分麦田作为“中转站”,利用水渠连接不同的麦田,这样只要一片麦田能被灌溉,则与其连接的麦田也能被灌溉。 现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用(注意不是所有麦田之间都可以建立水渠)。请问灌溉所有麦田最少需要多少费用来修建水渠。
输入格式
输入的第一行包含两个正整数n, m,分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
接下来m行,每行包含三个整数a i, b i, c i,表示第a i片麦田与第b i片麦田之间可以建立一条水渠,所需要的费用为c i。
输出格式
输出一行,包含一个整数,表示灌溉所有麦田所需要的最小费用。
样例输入
4 4
1 2 1 2 3 4 2 4 2 3 4 3
样例输出
6
样例说明
建立以下三条水渠:麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。
评测用例规模与约定
前20%的评测用例满足:n≤5。
前40%的评测用例满足:n≤20。 前60%的评测用例满足:n≤100。 所有评测用例都满足:1≤n≤1000,1≤m≤100,000,1≤c i≤10,000。
这个题目只需要不进行优化的prim就可以满分了,不过ccf不资磁auto和范围for让我编译错误了好几次。
#include<iostream>
#include<vector> #define MAX_VERTEX 1010 #define MAX_EDGE 10010 struct node { int vertex; int edgeValue; }; std::vector<node> map[MAX_VERTEX]; int n,m; bool shouldContinue(int *kown, int n) { int i; for (i = 1; i <= n; i++) { if (kown[i] == false) return true; } return false; } int main(void) { int i; int a, b, c; int d[MAX_VERTEX]; int kown[MAX_VERTEX]; long long int totalValue=0; std::cin >> n >> m; for (i = 1; i <= n; i++) { kown[i] = false; d[i] = MAX_EDGE; } node aNode; for (i = 1; i <= m; i++) { std::cin >> a >> b >> c; aNode.vertex = b; aNode.edgeValue = c; map[a].push_back(aNode); aNode.vertex = a; map[b].push_back(aNode); } d[1] = 0; int min; int minNumber=MAX_EDGE; while (shouldContinue(kown,n)) { minNumber = MAX_EDGE; //找到当前d最小的结点 for (i = 1; i <= n; i++) { if (kown[i] == false && d[i] < minNumber) { minNumber = d[i]; min = i; } }
kown[min] = true;
std::vector<node>::iterator vertex; for (vertex = map[min].begin(); vertex != map[min].end();vertex++) { if(kown[vertex->vertex]==false) d[vertex->vertex] = (vertex->edgeValue > d[vertex->vertex] ? d[vertex->vertex] : vertex->edgeValue);//更新所有结点的d值 } }
for (i = 1; i <= n; i++)
totalValue += d[i];
std::cout << totalValue;
return 0;
} |
ccf 最优灌溉
最新推荐文章于 2020-09-05 11:59:54 发布