operator< for Board

[slidingtile] / sliding_tile_lib.cc

diff --git a/sliding_tile_lib.cc b/sliding_tile_lib.cc
index aee8380c421fe11036f4c9bc957cca971edf6eca..e4f2f12723591ac320a56813eb8b3b306eaace30 100644 (file)
--- a/sliding_tile_lib.cc
+++ b/sliding_tile_lib.cc
@@ -2,10 +2,12 @@
 
 #include <cstdlib>
 #include <istream>
+#include <memory>
 #include <ostream>
 #include <stdexcept>
+#include <vector>
 
-signed char adjacent[BOARD_SIZE][5] = {
+signed char Step::adjacent[BOARD_SIZE][5] = {
   1,   4,   -1,  -1,  -1,
   0,   2,   5,   -1,  -1,
   1,   3,   6,   -1,  -1,
@@ -47,6 +49,30 @@ bool Board::is_valid() const {
   return true;
 }
 
+bool Board::operator==(const Board& o) const {
+  for (int i = 0; i < BOARD_SIZE; i++) {
+    if (board[i] != o.board[i]) {
+      return false;
+    }
+  }
+  return true;
+}
+
+bool Board::operator!=(const Board& o) const {
+  return !operator==(o);
+}
+
+bool Board::operator<(const Board& o) const {
+  for (int i = 0; i < BOARD_SIZE; i++) {
+    if (board[i] < o.board[i]) {
+      return true;
+    } else if (board[i] > o.board[i]) {
+      return false;
+    }
+  }
+  return false;
+}
+
 std::istream& operator>>(std::istream& is, Board& board) {
   for (int i = 0; i < BOARD_SIZE; i++) {
     if (!is.good()) {
@@ -106,3 +132,31 @@ int Board::distance(const InvertedBoard& invo) const {
   }
   return dist;
 }
+
+std::vector<Step*> Step::successors(std::shared_ptr<Step> shared_this) {
+  std::vector<Step*> suc;
+  signed char hole_pos = board.hole();
+  for (int i = 0; adjacent[hole_pos][i] > 0; i++) {
+    suc.emplace_back(new Step{board, shared_this});
+    std::swap(suc.back()->board.board[hole_pos], suc.back()->board.board[adjacent[hole_pos][i]]);
+  }
+  return suc;
+}
+
+std::ostream& operator<<(std::ostream& os, const Step& step) {
+  if (step.prev != nullptr) {
+    os << *step.prev;
+    signed char this_hole = step.board.hole();
+    signed char prev_hole = step.prev->board.hole();
+    os << int(step.board.board[prev_hole]) << " ";
+    switch (this_hole - prev_hole) {
+      case -1:         os << "right";    break;
+      case  1:         os << "left";     break;
+      case -BOARD_DIM: os << "down";     break;
+      case BOARD_DIM:  os << "up";       break;
+      default:         os << "somehow!"; break;
+    }
+    os << std::endl;
+  }
+  return os;
+}