题目描述:
题解:
这道题是学习回溯算法的时候刷到的,但其实感觉还是DFS的方法解决。
题目要求将所有被X包围的O改为O,被包围的部分不会属于边界,所以也就是将除了边界处O或与边界处O相连的部分外(相连的定义是上下左右四个方向),全部的O改为X。
1.定义一个DFS函数,输入为一个坐标(x,y),如果该位置对应的是O则将其标记为M,然后对其上下左右四个位置调用DFS。
2.对输入board的四个边界的每个位置调用DFS函数。
3.此时已经将所有不需要更改的O标志为了M,然后再对board的每个位置依次判断,将M改为O,O改为X即可。
class Solution(object): def solve(self, board): m = len(board) n = len(board[0]) directions = [(0,-1),(0,1),(-1,0),(1,0)] def dfs(x,y): if 0<=x<m and 0<=y<n and board[x][y]=='O': board[x][y]='M' for dx,dy in directions: newx = x+dx newy = y+dy dfs(newx,newy) for i in range(m): dfs(i,0) dfs(i,n-1) for j in range(n): dfs(0,j) dfs(m-1,j) for i in range(m): for j in range(n): if board[i][j]=='O': board[i][j]='X' if board[i][j]=='M': board[i][j]='O' return board