Ejemplo n.º 1
0
    def __init__(self, n, path="input.txt"):
        self.size = n
        with open(path, 'r') as f:
            lines = f.readlines()

            self.our_player = int(lines[0])
            self.other_player = 3 - self.our_player
            self.cur_step_num = None
            # self.good_moves = [Move(Point(1, 1)),
            #                    Move(Point(1, 2)),
            #                    Move(Point(1, 3)),
            #                    Move(Point(2, 1)),
            #                    Move(Point(2, 3)),
            #                    Move(Point(3, 1)),
            #                    Move(Point(3, 2)),
            #                    Move(Point(3, 3))]
            self.good_moves = [
                Move(Point(1, 1)),
                Move(Point(1, 3)),
                Move(Point(3, 1)),
                Move(Point(3, 3))
            ]

            previous_board_arr = []
            current_board_arr = []
            black_stones = 0
            white_stones = 0

            for line in lines[1:self.size + 1]:
                row = [int(x) for x in line.rstrip('\n')]

                previous_board_arr.append(row)

            for line in lines[self.size + 1:2 * self.size + 1]:
                row = []
                for x in line.rstrip('\n'):
                    row.append(int(x))
                    if int(x) == BLACK:
                        black_stones += 1
                    elif int(x) == WHITE:
                        white_stones += 1
                current_board_arr.append(row)

            previous_board = Board(self.size, previous_board_arr)
            current_board = Board(self.size, current_board_arr)
            self.board = current_board
            self.previous_state = State(previous_board, None,
                                        self.other_player, None)
            self.active_state = State(current_board, self.previous_state,
                                      self.our_player, None)
            self.set_step_num(black_stones, white_stones)
Ejemplo n.º 2
0
 def __init__(self, n):
     self.size = n
     self.board = Board(self.size)