Create surrounded_regions.py

Author: avidLearnerInProgress

surrounded_regions.py

@@ -0,0 +1,45 @@
+class Solution:
+    def solve(self, board: List[List[str]]) -> None:
+        n = len(board)
+        m = len(board[0])
+        
+        # If board have less than 3 size in any direction: nothing to do, because all cells located on borders
+        if n < 3 or m < 3:            
+            return
+        
+        # DFS to look for the next 'O' cell upper, lower, to the right and to the left of current coordinates
+        # If 'O' cell is found, recursevly mark this cell as 'R' which is mean REACHED
+        def dfs(row: int, col: int) -> None:
+            board[row][col] = 'R'
+            if row > 0 and board[row - 1][col] == 'O':
+                dfs(row - 1, col)
+            if row < n - 1 and board[row + 1][col] == 'O':
+                dfs(row + 1, col)
+            if col > 0 and board[row][col - 1] == 'O':
+                dfs(row, col - 1)
+            if col < m - 1 and board[row][col + 1] == 'O':
+                dfs(row, col + 1)
+        
+        # Go and check left and right borders of the board
+        for row in range(n):
+            if board[row][0] == 'O':
+                dfs(row, 0)
+            if board[row][m - 1] == 'O':
+                dfs(row, m - 1)
+        
+        # Same for check up and down borders of the board
+        # Since corners (0,0) and (n - 1, m - 1) where checked in previous cycle, skip them in this one
+        for col in range(1, m - 1):
+            if board[0][col] == 'O':
+                dfs(0, col)
+            if board[n - 1][col] == 'O':
+                dfs(n - 1, col)
+        
+        # Follow through the whole board and flip all 'R' cells back into 'O' and all 'O' cell to 'X'
+        # since they're unreacheable from the board located 'O' cell if any
+        for row in range(n):
+            for col in range(m):
+                if board[row][col] == 'O':
+                    board[row][col] = 'X'
+                elif board[row][col] == 'R':
+                    board[row][col] = 'O'