Tranvick

Prima

Oct 12th, 2013
158
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.91 KB | None | 0 0
  1. //Минимальное остовное дерево. Алгоритм Прима.
  2. #include <iostream>
  3. using namespace std;
  4.  
  5. const int N = 5555;
  6. const int M = 111111;
  7. const int INF = 1000000000;
  8. int mn[N], es[M], ev[M], first[N], next[M], c, n, m;
  9. bool b[N];
  10.  
  11. void add(int x, int y, int z) {
  12.     next[++c] = first[x]; first[x] = c;
  13.     es[c] = y; ev[c] = z;
  14. }
  15.  
  16. int prim() {
  17.     int ans = 0;// Вес минимального остова.
  18.     for (int i = 2; i <= n; ++i) mn[i] = INF;
  19.     for (int i = 1; i <= n; ++i) {
  20.         int v = -1;
  21.         for (int j = 1; j <= n; ++j)
  22.             if (!b[j] && (v == -1 || mn[j] < mn[v])) v = j;
  23.         b[v] = 1;
  24.         ans += mn[v];
  25.         for (int h = first[v]; h; h = next[h]) mn[es[h]] = min(ev[h], mn[es[h]]);
  26.     }
  27.     return ans;
  28. }
  29.  
  30. int main() {
  31.     cin >> n >> m;
  32.     for (int i = 0; i < m; ++i) {
  33.         int x, y, z;
  34.         cin >> x >> y >> z;
  35.         add(x, y, z);
  36.         add(y, x, z);
  37.     }
  38.     cout << prim();
  39.     return 0;
  40. }
Advertisement
Add Comment
Please, Sign In to add comment