题目如下:
In an
n*n
grid, there is a snake that spans 2 cells and starts moving from the top left corner at(0, 0)
and(0, 1)
. The grid has empty cells represented by zeros and blocked cells represented by ones. The snake wants to reach the lower right corner at(n-1, n-2)
and(n-1, n-1)
.In one move the snake can:
- Move one cell to the right if there are no blocked cells there. This move keeps the horizontal/vertical position of the snake as it is.
- Move down one cell if there are no blocked cells there. This move keeps the horizontal/vertical position of the snake as it is.
- Rotate clockwise if it's in a horizontal position and the two cells under it are both empty. In that case the snake moves from
(r, c)
and(r, c+1)
to(r, c)
and(r+1, c)
.- Rotate counterclockwise if it's in a vertical position and the two cells to its right are both empty. In that case the snake moves from
(r, c)
and(r+1, c)
to(r, c)
and(r, c+1)
.Return the minimum number of moves to reach the target.
If there is no way to reach the target, return
-1
.Example 1:
Input: grid = [[0,0,0,0,0,1], [1,1,0,0,1,0], [0,0,0,0,1,1], [0,0,1,0,1,0], [0,1,1,0,0,0], [0,1,1,0,0,0]] Output: 11 Explanation: One possible solution is [right, right, rotate clockwise, right, down, down, down, down, rotate counterclockwise, right, down].Example 2:
Input: grid = [[0,0,1,1,1,1], [0,0,0,0,1,1], [1,1,0,0,0,1], [1,1,1,0,0,1], [1,1,1,0,0,1], [1,1,1,0,0,0]] Output: 9Constraints:
2 <= n <= 100
0 <= grid[i][j] <= 1
- It is guaranteed that the snake starts at empty cells.
解题思路:典型的BFS题目。特别要注意的是蛇在水平/垂直方向是可以平移的。比如当前所在的左边是(0,0)(0,1),可以平移到(1,0),(1,1)。
代码如下:
class Solution(object): def minimumMoves(self, grid): """ :type grid: List[List[int]] :rtype: int """ res = float('inf') queue = [(0,0,0,1,0)] dic = {} dic[(0,0,0,1)] = 0 while len(queue) > 0: tx,ty,hx,hy,count = queue.pop(0) #print tx,ty,hx,hy,count if hx == len(grid) - 1 == hy and tx == len(grid)-1 and ty == len(grid) - 2: res = min(res,count) continue if tx == hx and ty < hy: #head to right if hy + 1 < len(grid) and grid[hx][hy+1] == 0: key = (tx,ty+1,hx,hy+1) if key not in dic or dic[key] > count + 1: queue.append((tx,ty+1,hx,hy+1,count+1)) dic[key] = count + 1 if hx + 1 < len(grid) and grid[tx+1][ty] == 0 and grid[hx+1][hy] == 0: key = (tx, ty, hx+1, ty) if key not in dic or dic[key] > count + 1: queue.append((tx, ty, hx+1, ty, count + 1)) dic[key] = count + 1 key = (tx+1,ty,hx+1,hy) if key not in dic or dic[key] > count + 1: queue.append((tx+1,ty,hx+1,hy, count + 1)) dic[key] = count + 1 elif tx < hx and ty == hy: #head to down if hx + 1 < len(grid) and grid[hx+1][hy] == 0: key = (tx+1,ty,hx+1,hy) if key not in dic or dic[key] > count + 1: queue.append((tx+1,ty,hx+1,hy,count+1)) dic[key] = count + 1 if hy + 1 < len(grid) and grid[hx][hy+1] == 0 and grid[tx][ty+1] == 0: key = tx,ty,tx,ty+1 if key not in dic or dic[key] > count + 1: queue.append((tx,ty,tx,ty+1,count+1)) dic[key] = count + 1 key = tx, ty+1, tx, ty + 1 if key not in dic or dic[key] > count + 1: queue.append((tx,ty+1,hx,hy+1,count+1)) dic[key] = count + 1 return res if res != float('inf') else -1