1. 程式人生 > >最小優先佇列+prim

最小優先佇列+prim

#include <iostream>
#include <cstdlib>
#include <queue>
 
using namespace std;

class Edge {
  public:
    int beg;
    int end;
    int val;

  friend bool operator<(const Edge &, const Edge &);
};

bool operator<(const Edge &e1, const Edge &e2) {
  return e1.val > e2.val;    // 最小優先佇列
}

const int MAX_N = 100;       // 頂點的最大數量
bool visit[MAX_N];
int value[MAX_N][MAX_N];
int link[MAX_N][MAX_N]; 

int main() {
  int n, m;
  int p, q, v;
  int totVisit, curVertex;
  int ans;
  Edge tmp;
  priority_queue<Edge> edgePQ;

  cin >> n >> m;  // n個頂點(標號0 -- n-1,m條邊)

  for (int i = 0; i != n; ++i) {
    visit[i] = false;
    link[i][0] = 1;
  }
  for (int i = 0; i != m; ++i) {
    cin >> p >> q >> v;

    value[p][q] = value[q][p] = v;
    link[p][link[p][0]++] = q;
    link[q][link[q][0]++] = p;
  }

  visit[0] = true;
  totVisit = 1;
  curVertex = 0;
  ans = 0;
  while (totVisit++ < n) {
    for (int i = 1; i != link[curVertex][0]; ++i) {
      if (!visit[link[curVertex][i]]) {
        tmp.beg = curVertex;
        tmp.end = link[curVertex][i];
        tmp.val = value[tmp.beg][tmp.end];

        edgePQ.push(tmp);
      }
    }

    while (!edgePQ.empty() && visit[edgePQ.top().end]) {
      edgePQ.pop();
    }
    tmp = edgePQ.top();
    visit[tmp.end] = true;
    ans += tmp.val;
    curVertex = tmp.end;

    edgePQ.pop();
  }

  cout << ans << endl;

  return EXIT_SUCCESS;
}

需要注意的地方就是,由於用的是STL中的priority_queue,所以要過載<操作符。並且其預設的是最大優先佇列,所以
<操作符的時候要返回的是>的值( >= 會出錯,原因我暫時不是很清楚)。
演算法時間複雜度近似於O(ElgV),基本達到要求。用最小優先佇列要比不用而直接每次全部點遍歷地找最小的邊效率高出
多(演算法導論那本書上說用斐波拉契堆實現的話執行時間可以減少到O(E + VlgV))。