LeetCode 657. 机器人能否返回原点 | Python
657. 机器人能否返回原点
题目来源:力扣(LeetCode)https://leetcode-cn.com/problems/robot-return-to-origin
题目
在二维平面上,有一个机器人从原点 (0, 0) 开始。给出它的移动顺序,判断这个机器人在完成移动后是否在 (0, 0) 处结束。
移动顺序由字符串表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R(右),L(左),U(上)和 D(下)。如果机器人在完成所有动作后返回原点,则返回 true。否则,返回 false。
注意:机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次,“L” 将始终向左移动等。此外,假设每次移动机器人的移动幅度相同。
示例 1:
输入: "UD"
输出: true
解释:机器人向上移动一次,然后向下移动一次。所有动作都具有相同的幅度,因此它最终回到它开始的原点。因此,我们返回 true。
示例 2:
输入: "LL"
输出: false
解释:机器人向左移动两次。它最终位于原点的左侧,距原点有两次 “移动” 的距离。我们返回 false,因为它在移动结束时没有返回原点。
解题思路
思路:模拟
先看题目,题目要求,在给定移动顺序后,机器人从原点 (0, 0)
出发,移动完毕后是否能够回到原点。
其中,机器人可以移动范围为四个方位,以下字母分别对应四个方位:
- R(右)
- L(左)
- U(上)
- D(下)
最后注意部分也额外说明,机器人回到原点无论面向哪个方位都可以。
那么,我们的思路就是模拟机器人行走。在二维平面上。定义 x,y 轴,机器人从原点(0,0)出发。当指令为 R,U
表示机器人分别往 x,y 轴的正方向移动,L,D
表示机器人分别往 x,y 轴的负方向移动。图示如下:
现在,定义变量 x,y,遍历字符串 moves,根据字符串中的指令移动机器人,移动完毕,判断机器人是否回到原点。做法如下:
- 定义变量 x,y;
- 遇到以下指令,更新 x,y:
- 遇到 'R',x += 1;
- 遇到 'L',x -= 1;
- 遇到 'U',y += 1;
- 遇到 'D',y -= 1;
- 移动完毕,判断是否回到原点,也就是 x,y 是否都为 0。
具体代码实现如下。
代码实现
class Solution:
def judgeCircle(self, moves: str) -> bool:
# 定义变量 x,y
x = 0
y = 0
# 遍历 moves,获取每步移动指令
for move in moves:
# 对应 4 个方位指令,分别更新 x,y 的值
if move == 'R':
x += 1
elif move == 'L':
x -= 1
elif move == 'U':
y += 1
elif move == 'D':
y -= 1
# 如果 x,y 的值都为 0,表示机器人回到原点
if x == 0 and y == 0:
return True
return False
实现结果
欢迎关注
公众号 【书所集录】