example.cu

#include <cstdint>
#include <iostream>

#include "graph.hpp"
#include "mst.hpp"

using namespace std;

int main(int argc, char **argv) {
  while (true) {
    uint32_t vertices;
    cin >> vertices;

    if (vertices == 0)
      break;

    CSRGraphBuilder builder{vertices};

    uint32_t edges;
    cin >> edges;

    for (uint32_t i = 0; i < edges; i++) {
      uint32_t u, v, w;
      cin >> u >> v >> w;

      // map 0..1
      u--;
      v--;

      builder.addEdge(u, v, w);
      builder.addEdge(v, u, w);
    }
    auto graph = builder.build();
    auto *r = MST(graph);

    uint32_t mstPathCost = 0;
    for (uint32_t i = 0; i < graph.E; i++) {
      if (r[i] == true) {
        mstPathCost += graph.W[i];
      }
    }
    cout << mstPathCost << '\n';

    cudaFree(r);
  }

  return 0;
}