View difference between Paste ID: TM4efCJ6 and NwF6nTN6

 #!/usr/bin/python
 import itertools
 
-class Game:
+class Cube36:
-    def __init__(self):
+    def __init__(self, board):
-        self.possible_positions = [ [1,2,5,4,6,3], [5,3,6,1,4,2], [4,6,3,5,2,1], [2,1,4,3,5,6], [3,5,2,6,1,4], [6,4,1,2,3,5] ]
+        self.possible_positions = board
-        self.board = [ [ '*' for i in range(6) ] for j in range(6) ]
+        size = len(self.possible_positions[0])
-        self.color_permutations = list(itertools.permutations(['p', 'r', 'c', 'o', 'y', 'g']))
+        print "The board: "
-        #self.possible_positions = [ [1,2,3],[2,3,1],[3,1,2] ]
+        print print_matrix(self.possible_positions)
-        #self.board = [ [ '*' for i in range(3) ] for j in range(3) ]
+        self.board = [ [ '*' for i in range(size) ] for j in range(size) ]
-        #self.color_permutations = list(itertools.permutations(['p','r','c']))
+        colors = ['p','r','c','o','y','g','m','t','b'][:size]
         self.color_permutations = list(itertools.permutations(colors))
 
-        output = ""
+
-        for i in self.board:
+        return print_matrix(self.board)
-            output = output + " ".join(i) + "\n"
+
-        return output.strip()
+
         if n == len(self.possible_positions[0]) + 1:
             print "Solution: "
             print self
             return True
         for colorcombo in self.color_permutations:
             colorcombo = list(colorcombo)
             if self.check_and_fill_table(colorcombo, n) == True:
                 self.solve(n+1)
-                return self.solve(n+1)
+
                 self.make_stars(n)
 
-                pass
+
-        print "lame."
+
             for colnum, value in enumerate(line):
                 if value == n:
                     self.board[linenum][colnum] = '*'
-            self.board[linenum][line.index(n)] = '*'
+
 
     def check_and_fill_table(self, colorcombo, n):
         for rownum, row in enumerate(self.possible_positions):
-            color = colorcombo.pop(0)
+            for colnum, value in enumerate(row):
-            if self.color_in_row(color, linenum):
+                if value == n:
-                return False
+                    color = colorcombo.pop(0)
-            if self.color_in_column(color, line.index(n)):
+                    if self.color_in_row(color, rownum):
-                return False
+                        return False
-            self.board[linenum][line.index(n)] = color
+                    if self.color_in_column(color, colnum):
                         return False
                     self.board[rownum][colnum] = color
         return True
-        if color in self.board[linenum]:
+
     def color_in_row(self, color, linenum):
         return color in self.board[linenum]
-    def color_in_column(self, color, column):
+
     def color_in_column(self, color, colnum):
-            if row[column] == color:
+
             if row[colnum] == color:
                 return True
 
-a = Game()
+def print_matrix(matrix):
-a.solve()
+    output = ""
     for row in matrix:
         output += "|"
         output += " ".join([str(col).center(2) for col in row])
         output += "|\n"
     return output.strip()
 
 
 # The board, as viewed by casual observer. (36 officers)
 #gameboard = [ [1,2,5,4,6,3],
 #              [5,3,6,1,4,2],
 #              [4,6,3,5,2,1],
 #              [2,1,4,3,5,6],
 #              [3,5,2,6,1,4],
 #              [6,4,1,2,3,5] ]
 
 # The board in reality.
 gameboard = [ [5,3,2,1,4,6],
               [4,1,5,2,5,3],
               [6,5,3,4,1,2],
               [1,2,6,3,6,4],
               [2,4,1,6,3,5],
               [3,6,4,5,2,1] ]
 
 # A 3x3 arrangement of officers. (test)
 #gameboard = [ [1, 2, 3],
 #              [2, 3, 1],
 #              [3, 1, 2] ]
 
 game = Cube36(gameboard)
 game.solve()