657. 机器人能否返回原点

在二维平面上,有一个机器人从原点 (0, 0) 开始。给出它的移动顺序,
判断这个机器人在完成移动后是否在 (0, 0) 处结束。
移动顺序由字符串表示。机器人的有效动作有 R(右),L(左),U(上)和 D(下)。
如果机器人在完成所有动作后返回原点,则返回 true。否则,返回 false。
注意:机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次,“L” 将始终向左移动等。
此外,假设每次移动机器人的移动幅度相同。

示例 1:
输入: "UD"
输出: true


 
 
思路:
1、设置两个计数器分别记录U和R的出现次数;
2、遍历字符串,遇到D则与U抵消,L与R抵消,最后计数器均为0则回到了原点。
 1 class Solution(object):
 2     def judgeCircle(self, moves):
 3         """
 4         :type moves: str
 5         :rtype: bool
 6         """
 7         if moves == "":
 8             return True
 9         # 设置两个计数器
10         numu = 0
11         numr = 0
12         # 遍历字符串,遇到D则与U抵消,L与R抵消,最后计数器均为0则回到了原点
13         for i in range(len(moves)):
14             if moves[i] == 'U':
15                 numu += 1
16             elif moves[i] == 'D':
17                 numu -= 1
18             elif moves[i] == 'R':
19                 numr += 1
20             else:
21                 numr -= 1
22         if numu == 0 and numr == 0:
23             return True
24         else:
25             return False
26 
27 
28 if __name__ == '__main__':
29     solution = Solution()
30     print(solution.judgeCircle("UD"))

 

 
 

你可能感兴趣的:(657. 机器人能否返回原点)