X-Git-Url: http://git.scottworley.com/slidingtile/blobdiff_plain/f3f55aff4e3346b297215050c2076c821fd60c83..3f2baedfbb6b08093ef6df37cbabc2ff15765b9f:/sliding_tile_lib.cc

diff --git a/sliding_tile_lib.cc b/sliding_tile_lib.cc
index b8ddc15..aee8380 100644
--- a/sliding_tile_lib.cc
+++ b/sliding_tile_lib.cc
@@ -1,6 +1,8 @@
 #include "sliding_tile_lib.h"
 
+#include <cstdlib>
 #include <istream>
+#include <ostream>
 #include <stdexcept>
 
 signed char adjacent[BOARD_SIZE][5] = {
@@ -22,7 +24,7 @@ signed char adjacent[BOARD_SIZE][5] = {
   11,  14,  -1,  -1,  -1,
 };
 
-bool Board::is_valid() {
+bool Board::is_valid() const {
   bool seen[BOARD_SIZE];
   for (int i = 0; i < BOARD_SIZE; i++) {
     seen[i] = false;
@@ -65,7 +67,17 @@ std::istream& operator>>(std::istream& is, Board& board) {
   return is;
 }
 
-signed char Board::hole() {
+std::ostream& operator<<(std::ostream& os, const Board& board) {
+  for (int i = 0; i < BOARD_SIZE; i++) {
+    if (i > 0) {
+      os << " ";
+    }
+    os << int(board.board[i]);
+  }
+  return os;
+}
+
+signed char Board::hole() const {
   for (int i = 0; i < BOARD_SIZE; i++) {
     if (board[i] == 0) {
       return i;
@@ -73,3 +85,24 @@ signed char Board::hole() {
   }
   throw std::runtime_error("Board with no hole");
 }
+
+InvertedBoard Board::invert() const {
+  InvertedBoard inv;
+  for (int i = 0; i < BOARD_SIZE; i++) {
+    inv.pos[board[i]] = i;
+  }
+  return inv;
+}
+
+int Board::distance(const Board& o) const {
+  return distance(o.invert());
+}
+
+int Board::distance(const InvertedBoard& invo) const {
+  int dist = 0;
+  for (int i = 0; i < BOARD_SIZE; i++) {
+    dist += std::abs(i % BOARD_DIM - invo.pos[board[i]] % BOARD_DIM) +
+            std::abs(i / BOARD_DIM - invo.pos[board[i]] / BOARD_DIM);
+  }
+  return dist;
+}