grprimm1.cpp

来自「经典c++程序的实现」· C++ 代码 · 共 44 行

// Prim's MST algorithm -- v^2 version
#include <stdio.h>
#include <iostream.h>
#include <assert.h>

#include "..\..\include\book.h"

#include "..\..\include\grmat.h"

void AddEdgetoMST(int v1, int v2) {
  cout << "Add edge " << v1 << " to " << v2 << "\n";
}

int minVertex(Graph&, int*);

#include "..\..\include\grprim1.c"

void main(int argc, char** argv) {
  Graph G;
  FILE *fid;
  int i;

  if (argc != 2) {
    cout << "Usage: grprimm1 <file>\n";
    exit(-1);
  }

  if ((fid = fopen(argv[1], "rt")) == NULL) {
    cout << "Unable to open file |" << argv[1] << "|\n";
    exit(-1);
  }

  if (!createGraph(G, fid)) {
    cout << "Unable to create graph\n";
    exit(-1);
  }

  // Initialize Mark array
  for (i=0; i<G.n(); i++)
    G.Mark[i] = UNVISITED;

  Prim(G, 0);
}