| 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: Hit an 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 is adjacent to both players | |
| 70 | if len(owners) == 2: | |
| 71 | return ('', territory_coords) | |
| 72 | ||
| 73 | # Edge Case: Territory is adjacent to one player | |
| 74 | if len(owners) == 1: | |
| 75 | return (list(owners)[0], territory_coords) | |
| 76 | ||
| 77 | # Edge Case: Territory is not adjacent to any player (neutral territory) | |
| 78 | return ('', territory_coords) | |
| 79 | # Handled Edge Cases: Invalid coordinates (negative), Invalid coordinates (out of bounds), Coordinate is not empty, Out of bounds or already visited, Hit a stone, Hit an empty space, Territory is adjacent to both players, Territory is adjacent to one player, Territory is not adjacent to any player (neutral territory) | |
| 27 | 80 | |
| 28 | 81 | def territories(self) -> dict[str, set[tuple[int, int]]]: |
| 29 | 82 | """Find the owners and the territories of the whole board |
| 36 | 89 | , i.e. "W", "B", "". The value being a set |
| 37 | 90 | of coordinates owned by the owner. |
| 38 | 91 | """ |
| 39 | pass | |
| 92 | result = {'W': set(), 'B': set(), '': set()} | |
| 93 | visited_global = set() | |
| 94 | ||
| 95 | for y in range(self.rows): | |
| 96 | for x in range(self.cols): | |
| 97 | # Edge Case: Skip non-empty positions and already visited positions | |
| 98 | if (x, y) in visited_global or self.board[y][x] != ' ': | |
| 99 | continue | |
| 100 | ||
| 101 | owner, territory_coords = self.territory(x, y) | |
| 102 | result[owner].update(territory_coords) | |
| 103 | visited_global.update(territory_coords) | |
| 104 | ||
| 105 | return result | |
| 106 | # Handled Edge Cases: Skip non-empty positions and already visited positions |
| 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.
