diff options
Diffstat (limited to 'voerbak/main.py')
| -rw-r--r-- | voerbak/main.py | 46 |
1 files changed, 23 insertions, 23 deletions
diff --git a/voerbak/main.py b/voerbak/main.py index d802603..2ca473e 100644 --- a/voerbak/main.py +++ b/voerbak/main.py @@ -12,62 +12,62 @@ class bord: self.width = w self.height = h self.board = [ - [EMPTY for x in range(self.width)] for u in range(self.height) + [EMPTY for x in range(self.width)] for u in range(self.height) ] def print(self): print( - "\n".join( - [ - " ".join(self.board[y]) - for y in range(len(self.board) - 1, -1, -1) - ] - ) + "\n".join( + [ + " ".join(self.board[y]) + for y in range(len(self.board) - 1, -1, -1) + ] + ) ) def outside_board(self, coords): return coords[0] < 0 or \ - coords[1] < 0 or \ - coords[0] > self.height - 1 or \ - coords[1] > self.width - 1 + coords[1] < 0 or \ + coords[0] > self.height - 1 or \ + coords[1] > self.width - 1 def recursive_solve(self, coords, check_for, direction, current_length): new_position = (coords[0] + direction[0], coords[1] + direction[1]) if self.outside_board(new_position) or self.board[new_position[0]][ - new_position[1]] != check_for: + new_position[1]] != check_for: return current_length else: return self.recursive_solve( - new_position, check_for, direction, current_length + 1 + new_position, check_for, direction, current_length + 1 ) def check_win(self, coords): directions = [ - (1, 0), (1, 1), (0, 1), (-1, 1), (-1, 0), (-1, -1), (0, -1), - (1, -1) + (1, 0), (1, 1), (0, 1), (-1, 1), (-1, 0), (-1, -1), (0, -1), + (1, -1) ] values = list() for direction in directions: values.append( - self.recursive_solve( - coords, self.board[coords[0]][coords[1]], direction, 0 - ) + self.recursive_solve( + coords, self.board[coords[0]][coords[1]], direction, 0 + ) ) joined_directions = [ - values[0] + values[4], values[1] + values[5], - values[2] + values[6], values[3] + values[7] + values[0] + values[4], values[1] + values[5], + values[2] + values[6], values[3] + values[7] ] won = any(i >= 3 for i in joined_directions) if won: for i, value in enumerate(joined_directions): if value >= 3: start_pos = ( - coords[0] + directions[i][0] * values[i], - coords[1] + directions[i][1] * values[i], + coords[0] + directions[i][0] * values[i], + coords[1] + directions[i][1] * values[i], ) end_pos = ( - coords[0] + directions[i + 4][0] * values[i + 4], - coords[1] + directions[i + 4][1] * values[i + 4], + coords[0] + directions[i + 4][0] * values[i + 4], + coords[1] + directions[i + 4][1] * values[i + 4], ) print(start_pos, end_pos) |