Question

489. Robot Room Cleaner

中文文档

Description

You are controlling a robot that is located somewhere in a room. The room is modeled as an m x n binary grid where 0 represents a wall and 1 represents an empty slot.

The robot starts at an unknown location in the room that is guaranteed to be empty, and you do not have access to the grid, but you can move the robot using the given API Robot.

You are tasked to use the robot to clean the entire room (i.e., clean every empty cell in the room). The robot with the four given APIs can move forward, turn left, or turn right. Each turn is 90 degrees.

When the robot tries to move into a wall cell, its bumper sensor detects the obstacle, and it stays on the current cell.

Design an algorithm to clean the entire room using the following APIs:

interface Robot {
  // returns true if next cell is open and robot moves into the cell.
  // returns false if next cell is obstacle and robot stays on the current cell.
  boolean move();

  // Robot will stay on the same cell after calling turnLeft/turnRight.
  // Each turn will be 90 degrees.
  void turnLeft();
  void turnRight();

  // Clean the current cell.
  void clean();
}

Note that the initial direction of the robot will be facing up. You can assume all four edges of the grid are all surrounded by a wall.

 

Custom testing:

The input is only given to initialize the room and the robot's position internally. You must solve this problem "blindfolded". In other words, you must control the robot using only the four mentioned APIs without knowing the room layout and the initial robot's position.

 

Example 1:

Input: room = [[1,1,1,1,1,0,1,1],[1,1,1,1,1,0,1,1],[1,0,1,1,1,1,1,1],[0,0,0,1,0,0,0,0],[1,1,1,1,1,1,1,1]], row = 1, col = 3
Output: Robot cleaned all rooms.
Explanation: All grids in the room are marked by either 0 or 1.
0 means the cell is blocked, while 1 means the cell is accessible.
The robot initially starts at the position of row=1, col=3.
From the top left corner, its position is one row below and three columns right.

Example 2:

Input: room = [[1]], row = 0, col = 0
Output: Robot cleaned all rooms.

 

Constraints:

• m == room.length
• n == room[i].length
• 1 <= m <= 100
• 1 <= n <= 200
• room[i][j] is either 0 or 1.
• 0 <= row < m
• 0 <= col < n
• room[row][col] == 1
• All the empty cells can be visited from the starting position.

Solutions

Solution 1

# """
# This is the robot's control interface.
# You should not implement it, or speculate about its implementation
# """
# class Robot:
#  def move(self):
#    """
#    Returns true if the cell in front is open and robot moves into the cell.
#    Returns false if the cell in front is blocked and robot stays in the current cell.
#    :rtype bool
#    """
#
#  def turnLeft(self):
#    """
#    Robot will stay in the same cell after calling turnLeft/turnRight.
#    Each turn will be 90 degrees.
#    :rtype void
#    """
#
#  def turnRight(self):
#    """
#    Robot will stay in the same cell after calling turnLeft/turnRight.
#    Each turn will be 90 degrees.
#    :rtype void
#    """
#
#  def clean(self):
#    """
#    Clean the current cell.
#    :rtype void
#    """


class Solution:
  def cleanRoom(self, robot):
    """
    :type robot: Robot
    :rtype: None
    """

    def dfs(i, j, d):
      vis.add((i, j))
      robot.clean()
      for k in range(4):
        nd = (d + k) % 4
        x, y = i + dirs[nd], j + dirs[nd + 1]
        if (x, y) not in vis and robot.move():
          dfs(x, y, nd)
          robot.turnRight()
          robot.turnRight()
          robot.move()
          robot.turnRight()
          robot.turnRight()
        robot.turnRight()

    dirs = (-1, 0, 1, 0, -1)
    vis = set()
    dfs(0, 0, 0)

/**
 * // This is the robot's control interface.
 * // You should not implement it, or speculate about its implementation
 * interface Robot {
 *   // Returns true if the cell in front is open and robot moves into the cell.
 *   // Returns false if the cell in front is blocked and robot stays in the current cell.
 *   public boolean move();
 *
 *   // Robot will stay in the same cell after calling turnLeft/turnRight.
 *   // Each turn will be 90 degrees.
 *   public void turnLeft();
 *   public void turnRight();
 *
 *   // Clean the current cell.
 *   public void clean();
 * }
 */

class Solution {
  private int[] dirs = {-1, 0, 1, 0, -1};
  private Set<List<Integer>> vis = new HashSet<>();
  private Robot robot;

  public void cleanRoom(Robot robot) {
    this.robot = robot;
    dfs(0, 0, 0);
  }

  private void dfs(int i, int j, int d) {
    robot.clean();
    vis.add(List.of(i, j));
    for (int k = 0; k < 4; ++k) {
      int nd = (d + k) % 4;
      int x = i + dirs[nd], y = j + dirs[nd + 1];
      if (!vis.contains(List.of(x, y)) && robot.move()) {
        dfs(x, y, nd);
        robot.turnRight();
        robot.turnRight();
        robot.move();
        robot.turnRight();
        robot.turnRight();
      }
      robot.turnRight();
    }
  }
}

/**
 * // This is the robot's control interface.
 * // You should not implement it, or speculate about its implementation
 * class Robot {
 *   public:
 *   // Returns true if the cell in front is open and robot moves into the cell.
 *   // Returns false if the cell in front is blocked and robot stays in the current cell.
 *   bool move();
 *
 *   // Robot will stay in the same cell after calling turnLeft/turnRight.
 *   // Each turn will be 90 degrees.
 *   void turnLeft();
 *   void turnRight();
 *
 *   // Clean the current cell.
 *   void clean();
 * };
 */

class Solution {
public:
  void cleanRoom(Robot& robot) {
    int dirs[5] = {-1, 0, 1, 0, -1};
    set<pair<int, int>> vis;
    function<void(int, int, int)> dfs = [&](int i, int j, int d) {
      robot.clean();
      vis.insert({i, j});
      for (int k = 0; k < 4; ++k) {
        int nd = (d + k) % 4;
        int x = i + dirs[nd], y = j + dirs[nd + 1];
        if (!vis.count({x, y}) && robot.move()) {
          dfs(x, y, nd);
          robot.turnRight();
          robot.turnRight();
          robot.move();
          robot.turnRight();
          robot.turnRight();
        }
        robot.turnRight();
      }
    };
    dfs(0, 0, 0);
  }
};

/**
 * // This is the robot's control interface.
 * // You should not implement it, or speculate about its implementation
 * type Robot struct {
 * }
 *
 * // Returns true if the cell in front is open and robot moves into the cell.
 * // Returns false if the cell in front is blocked and robot stays in the current cell.
 * func (robot *Robot) Move() bool {}
 *
 * // Robot will stay in the same cell after calling TurnLeft/TurnRight.
 * // Each turn will be 90 degrees.
 * func (robot *Robot) TurnLeft() {}
 * func (robot *Robot) TurnRight() {}
 *
 * // Clean the current cell.
 * func (robot *Robot) Clean() {}
 */

func cleanRoom(robot *Robot) {
  vis := map[[2]int]bool{}
  dirs := [5]int{-1, 0, 1, 0, -1}
  var dfs func(int, int, int)
  dfs = func(i, j, d int) {
    vis[[2]int{i, j}] = true
    robot.Clean()
    for k := 0; k < 4; k++ {
      nd := (d + k) % 4
      x, y := i+dirs[nd], j+dirs[nd+1]
      if !vis[[2]int{x, y}] && robot.Move() {
        dfs(x, y, nd)
        robot.TurnRight()
        robot.TurnRight()
        robot.Move()
        robot.TurnRight()
        robot.TurnRight()
      }
      robot.TurnRight()
    }
  }
  dfs(0, 0, 0)
}

/**
 * class Robot {
 *    // Returns true if the cell in front is open and robot moves into the cell.
 *    // Returns false if the cell in front is blocked and robot stays in the current cell.
 *     move(): boolean {}
 *
 *    // Robot will stay in the same cell after calling turnLeft/turnRight.
 *    // Each turn will be 90 degrees.
 *     turnRight() {}
 *
 *    // Robot will stay in the same cell after calling turnLeft/turnRight.
 *    // Each turn will be 90 degrees.
 *     turnLeft() {}
 *
 *     // Clean the current cell.
 *     clean(): {}
 * }
 */

function cleanRoom(robot: Robot) {
  const dirs = [-1, 0, 1, 0, -1];
  const vis = new Set<string>();
  const dfs = (i: number, j: number, d: number) => {
    vis.add(`${i}-${j}`);
    robot.clean();
    for (let k = 0; k < 4; ++k) {
      const nd = (d + k) % 4;
      const [x, y] = [i + dirs[nd], j + dirs[nd + 1]];
      if (!vis.has(`${x}-${y}`) && robot.move()) {
        dfs(x, y, nd);
        robot.turnRight();
        robot.turnRight();
        robot.move();
        robot.turnRight();
        robot.turnRight();
      }
      robot.turnRight();
    }
  };
  dfs(0, 0, 0);
}