poj 1258 Agri-Net 解题报告

题目链接：http://poj.org/problem?id=1258

题目意思：给出 n 个 farm，每个farm 之间通过一定数量的fiber 相连，问使得所有farm 直接或间接连通的 最少 fiber 数是多少。

赤裸裸的最小生成树，用prim做的。

有个地方写错，wa 了 几次。

 #include <iostream>
 #include <cstdio>
 using namespace std;

 const int farm_num =  + ;
 const int INF = 1e9;
 int farm[farm_num][farm_num];
 int dist[farm_num], vis[farm_num];
 int ans, N;

 void prim()
 {
     int k;
     for (int i = ; i <= N; i++)
     {
         vis[i] = ;
         dist[i] = INF;
     }
     // 假设从编号为 1 的farm 出发
     dist[] = , ans = ;
     for (int i = ; i <= N; i++)
     {
         int tmp = INF;
         for (int j = ; j <= N; j++)
         {
             if (!vis[j] && dist[j] < tmp)
             {
                 tmp = dist[j];
                 k = j;
             }
         }
         vis[k] = ;
         ans += tmp;
         for (int j = ; j <= N; j++)
         {
             if (!vis[j] && dist[j] > farm[k][j])  // 一开始将dist[j] > farm[k][j]直接写成farm[k][j]了= =
                 dist[j] = farm[k][j];
         }
     }
 }

 int main()
 {
     while (scanf("%d", &N) != EOF)
     {
         for (int i = ; i <= N; i++)
         {
             for (int j = ; j <= N; j++)
                 scanf("%d", &farm[i][j]);
         }
         prim();
         printf("%d\n", ans);
     }
     return ;
 }