Question

657. 机器人能否返回原点

English Version

题目描述

在二维平面上，有一个机器人从原点 (0, 0) 开始。给出它的移动顺序，判断这个机器人在完成移动后是否在 (0, 0) 处结束。

移动顺序由字符串 moves 表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R（右），L（左），U（上）和 D（下）。

如果机器人在完成所有动作后返回原点，则返回 true。否则，返回 false。

注意：机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次，“L” 将始终向左移动等。此外，假设每次移动机器人的移动幅度相同。

 

示例 1:

输入: moves = "UD"
输出: true
解释：机器人向上移动一次，然后向下移动一次。所有动作都具有相同的幅度，因此它最终回到它开始的原点。因此，我们返回 true。

示例 2:

输入: moves = "LL"
输出: false
解释：机器人向左移动两次。它最终位于原点的左侧，距原点有两次 “移动” 的距离。我们返回 false，因为它在移动结束时没有返回原点。

 

提示:

• 1 <= moves.length <= 2 * 104
• moves 只包含字符 'U', 'D', 'L' 和 'R'

解法

方法一

class Solution:
  def judgeCircle(self, moves: str) -> bool:
    x = y = 0
    for c in moves:
      if c == 'R':
        x += 1
      elif c == 'L':
        x -= 1
      elif c == 'U':
        y += 1
      elif c == 'D':
        y -= 1
    return x == 0 and y == 0

class Solution {
  public boolean judgeCircle(String moves) {
    int x = 0, y = 0;
    for (int i = 0; i < moves.length(); ++i) {
      char c = moves.charAt(i);
      if (c == 'R')
        ++x;
      else if (c == 'L')
        --x;
      else if (c == 'U')
        ++y;
      else if (c == 'D')
        --y;
    }
    return x == 0 && y == 0;
  }
}

function judgeCircle(moves: string): boolean {
  let x = 0,
    y = 0;
  const dir = {
    R: [1, 0],
    L: [-1, 0],
    U: [0, 1],
    D: [0, -1],
  };
  for (let u of moves) {
    const [dx, dy] = dir[u];
    x += dx;
    y += dy;
  }
  return !x && !y;
}