POJ 1258 Agri-Net(最小生成樹 Prim 模版題)
阿新 • • 發佈:2019-02-04
大意:新鎮長競選宣言就是將網路帶到每一個農場,給出農場個數,兩兩之間建光纜的耗費,求所有都聯通的最小耗費。
思路:最小生成樹,因為邊比較稠密,用Prim做。
PS;對於比較稠密的圖,用Prim,對於比較稀疏的圖,用 Kruskal。Kruskal是找邊的過程,稀疏的話會比較快。
#include <stdio.h> #include <string.h> #define INF 0x3f3f3f3f int dis[110]; int Map[110][110]; int n; int Ans; int min(int a, int b) { return a > b ? b : a; } void Prim() { int Min_ele, Min_node; memset(dis, INF, sizeof(dis)); Ans = 0; int r = 1; for(int i = 1; i < n; i++) { dis[r] = -1; Min_ele = INF; for(int j = 1; j <= n; j++) { if(dis[j] >= 0) { dis[j] = min(dis[j], Map[r][j]); if(dis[j] < Min_ele) { Min_ele = dis[j]; Min_node = j; } } } r = Min_node; Ans += Min_ele; } } void Solve() { while(~scanf("%d", &n)) { memset(Map, 0, sizeof(Map)); for(int i = 1; i <= n; i++) { for(int j = 1; j <= n; j++) { scanf("%d", &Map[i][j]); } } Prim(); printf("%d\n", Ans); } } int main() { Solve(); return 0; }