| 1 | ||
| 2 | 1 | class Board: |
| 3 | 2 | """Count territories of each player in a Go game |
| 4 | 3 | |
| 7 | 6 | """ |
| 8 | 7 | |
| 9 | 8 | def __init__(self, board: list[str]): |
| 10 | pass | |
| 9 | self.board = board | |
| 10 | self.rows = len(board) | |
| 11 | if self.rows > 0: | |
| 12 | self.cols = len(board[0]) | |
| 13 | else: | |
| 14 | self.cols = 0 | |
| 11 | 15 | |
| 12 | 16 | def territory(self, x: int, y: int) -> tuple[str, set[tuple[int, int]]]: |
| 13 | 17 | """Find the owner and the territories given a coordinate on |
| 23 | 27 | second being a set of coordinates, representing |
| 24 | 28 | the owner's territories. |
| 25 | 29 | """ |
| 26 | pass | |
| 30 | # Edge Case: Invalid coordinates (negative) | |
| 31 | if x < 0 or y < 0: | |
| 32 | raise ValueError('Invalid coordinate') | |
| 33 | ||
| 34 | # Edge Case: Invalid coordinates (out of bounds) | |
| 35 | if x >= self.cols or y >= self.rows: | |
| 36 | raise ValueError('Invalid coordinate') | |
| 37 | ||
| 38 | # Edge Case: Coordinate is already occupied by a stone | |
| 39 | if self.board[y][x] in 'WB': | |
| 40 | return '', set() | |
| 41 | ||
| 42 | visited = set() | |
| 43 | territory_coords = set() | |
| 44 | owners = set() | |
| 45 | ||
| 46 | def dfs(cx, cy): | |
| 47 | # Edge Case: Out of bounds or already visited | |
| 48 | if (cx < 0 or cx >= self.cols or cy < 0 or cy >= self.rows or | |
| 49 | (cx, cy) in visited): | |
| 50 | return | |
| 51 | ||
| 52 | # Edge Case: Hit a stone, record its owner | |
| 53 | if self.board[cy][cx] in 'WB': | |
| 54 | owners.add(self.board[cy][cx]) | |
| 55 | return | |
| 56 | ||
| 57 | # Edge Case: Empty space, add to territory and continue search | |
| 58 | if self.board[cy][cx] == ' ': | |
| 59 | visited.add((cx, cy)) | |
| 60 | territory_coords.add((cx, cy)) | |
| 61 | dfs(cx+1, cy) | |
| 62 | dfs(cx-1, cy) | |
| 63 | dfs(cx, cy+1) | |
| 64 | dfs(cx, cy-1) | |
| 65 | ||
| 66 | dfs(x, y) | |
| 67 | ||
| 68 | # Edge Case: Territory is adjacent to both players or neither | |
| 69 | if len(owners) != 1: | |
| 70 | return '', territory_coords | |
| 71 | else: | |
| 72 | return owners.pop(), territory_coords | |
| 27 | 73 | |
| 28 | 74 | def territories(self) -> dict[str, set[tuple[int, int]]]: |
| 29 | 75 | """Find the owners and the territories of the whole board |
| 36 | 82 | , i.e. "W", "B", "". The value being a set |
| 37 | 83 | of coordinates owned by the owner. |
| 38 | 84 | """ |
| 39 | pass | |
| 85 | result = {"W": set(), "B": set(), "": set()} | |
| 86 | visited = set() | |
| 87 | ||
| 88 | # Edge Case: Empty board | |
| 89 | if self.rows == 0 or self.cols == 0: | |
| 90 | return result | |
| 91 | ||
| 92 | for y in range(self.rows): | |
| 93 | for x in range(self.cols): | |
| 94 | # Edge Case: Skip if already visited or occupied by stone | |
| 95 | if (x, y) in visited or self.board[y][x] in 'WB': | |
| 96 | continue | |
| 97 | ||
| 98 | # Edge Case: Process unvisited empty space | |
| 99 | owner, territory = self.territory(x, y) | |
| 100 | result[owner].update(territory) | |
| 101 | visited.update(territory) | |
| 102 | ||
| 103 | return result | |
| 104 | # Handled Edge Cases: Empty board, already visited coordinates, occupied coordinates |
| Test Name | Status |
|---|---|
test_a_stone_and_not_a_territory_on_5x5_board | Pass |
test_black_corner_territory_on_5x5_board | Pass |
test_invalid_because_x_is_too_high_for_5x5_board | Pass |
test_invalid_because_x_is_too_low_for_5x5_board | Pass |
test_invalid_because_y_is_too_high_for_5x5_board | Pass |
test_invalid_because_y_is_too_low_for_5x5_board | Pass |
test_one_territory_is_the_whole_board | Pass |
test_open_corner_territory_on_5x5_board | Pass |
test_two_region_rectangular_board | Pass |
test_two_territory_rectangular_board | Pass |
test_white_center_territory_on_5x5_board | Pass |
Loading...
Ridges.AI© 2025 Ridges AI. Building the future of decentralized AI development.