| 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 not empty | |
| 39 | if self.board[y][x] != ' ': | |
| 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 | |
| 53 | if self.board[cy][cx] in 'WB': | |
| 54 | owners.add(self.board[cy][cx]) | |
| 55 | return | |
| 56 | ||
| 57 | # Edge Case: Empty space | |
| 58 | if self.board[cy][cx] == ' ': | |
| 59 | visited.add((cx, cy)) | |
| 60 | territory_coords.add((cx, cy)) | |
| 61 | # Explore neighbors | |
| 62 | dfs(cx+1, cy) | |
| 63 | dfs(cx-1, cy) | |
| 64 | dfs(cx, cy+1) | |
| 65 | dfs(cx, cy-1) | |
| 66 | ||
| 67 | dfs(x, y) | |
| 68 | ||
| 69 | # Edge Case: Territory belongs to one player | |
| 70 | if len(owners) == 1: | |
| 71 | return (owners.pop(), territory_coords) | |
| 72 | ||
| 73 | # Edge Case: Territory is neutral (touches both players or neither) | |
| 74 | return ('', territory_coords) | |
| 27 | 75 | |
| 28 | 76 | def territories(self) -> dict[str, set[tuple[int, int]]]: |
| 29 | 77 | """Find the owners and the territories of the whole board |
| 36 | 84 | , i.e. "W", "B", "". The value being a set |
| 37 | 85 | of coordinates owned by the owner. |
| 38 | 86 | """ |
| 39 | pass | |
| 87 | result = {"W": set(), "B": set(), "": set()} | |
| 88 | visited = set() | |
| 89 | ||
| 90 | # Edge Case: Empty board | |
| 91 | if self.rows == 0 or self.cols == 0: | |
| 92 | return result | |
| 93 | ||
| 94 | for y in range(self.rows): | |
| 95 | for x in range(self.cols): | |
| 96 | # Edge Case: Only process unvisited empty spaces | |
| 97 | if self.board[y][x] == ' ' and (x, y) not in visited: | |
| 98 | owner, territory_set = self.territory(x, y) | |
| 99 | result[owner].update(territory_set) | |
| 100 | visited.update(territory_set) | |
| 101 | ||
| 102 | return result | |
| 103 | # Handled Edge Cases: Empty board, unvisited empty spaces |
| 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.