#include "arc_factored.h"

#include <iostream>

using namespace std;

int main(int argc, char** argv) {
  // John saw Mary
  //   (H -> M)
  //   (1 -> 2) 20
  //   (1 -> 3) 3
  //   (2 -> 1) 20
  //   (2 -> 3) 30
  //   (3 -> 2) 0
  //   (3 -> 1) 11
  //   (0, 2) 10
  //   (0, 1) 9
  //   (0, 3) 9
  ArcFactoredForest af(3);
  af(1,2).edge_prob.logeq(20);
  af(1,3).edge_prob.logeq(3);
  af(2,1).edge_prob.logeq(20);
  af(2,3).edge_prob.logeq(30);
  af(3,2).edge_prob.logeq(0);
  af(3,1).edge_prob.logeq(11);
  af(0,2).edge_prob.logeq(10);
  af(0,1).edge_prob.logeq(9);
  af(0,3).edge_prob.logeq(9);
  SpanningTree tree;
  af.MaximumSpanningTree(&tree);
  return 0;
}