LeetCode 657. 机器人能否返回原点

题目

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

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

方法：暴力解法

若想回到原点，那么需要向左向右移动次数相同，向上向下移动次数相同

class Solution(object):
    def judgeCircle(self, moves):
        horizontal, vertical = 0, 0
        for i in range(len(moves)):
            if moves[i] == 'R':
                horizontal += 1
            elif moves[i] == 'L':
                horizontal -= 1
            elif moves[i] == 'U':
                vertical += 1
            elif moves[i] == 'D':
                vertical -= 1
        
        if horizontal == 0 and vertical == 0:
            return True
        else:
            return False