【力扣 657】机器人能否返回原点 C++题解(字符串+模拟)

45 阅读2分钟

在二维平面上,有一个机器人从原点 (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'


思路

定义两个变量xy,分别表示机器人在x轴和y轴上的位置。然后,函数计算moves中每种移动指令的数量,并根据移动指令的类型对xy进行相应的增减。

  • 如果移动指令是'R',那么机器人向右移动,x增加。
  • 如果移动指令是'L',那么机器人向左移动,x减少。
  • 如果移动指令是'U',那么机器人向上移动,y增加。
  • 如果移动指令是'D',那么机器人向下移动,y减少。

在处理完所有的移动指令后,如果xy都是0,那么机器人回到了原点,函数返回true。否则,函数返回false


AC代码

/*
 * @lc app=leetcode.cn id=657 lang=cpp
 *
 * [657] 机器人能否返回原点
 */

// @lc code=start
class Solution {
   public:
	bool judgeCircle(string moves) {
		int x, y;
		x = y = 0;
		x += count(moves.begin(), moves.end(), 'R');
		x -= count(moves.begin(), moves.end(), 'L');
		y += count(moves.begin(), moves.end(), 'U');
		y -= count(moves.begin(), moves.end(), 'D');
		return !(x || y);
	}
};
// @lc code=end