leetcode | 657.机器人能够返回原点

657.机器人能够返回原点

题目描述

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

题目链接

思路

起始时机器人的坐标为 (0,0)，在遍历完所有指令并对机器人进行移动之后，判断机器人的坐标是否为 (0,0) 即可。

具体来说，我们用两个变量 x 和 y 来表示机器人当前所在的坐标为 (x,y)，起始时 x=0，y=0。接下来我们遍历指令并更新机器人的坐标：
如果指令是 U，则令 y=y−1
如果指令是 D，则令 y=y+1
如果指令是 L，则令 x=x−1
如果指令是 R，则令 x=x+1
最后判断 (x,y) 是否为 (0,0) 即可。

代码

class Solution:
    def judgeCircle(self, moves: str) -> bool:
        if len(moves) % 2 != 0:
            return False

        pos_x = 0
        pos_y = 0

        for move in moves:
            if move == 'U':
                pos_y += 1
            elif move == 'D':
                pos_y -= 1
            elif move == 'R':
                pos_x += 1
            elif move == 'L':
                pos_x -= 1

        if pos_x == 0 and pos_y == 0:
            return True
        else:
            return False

    def judgeCircle1(self, moves: str) -> bool:
        if moves.count('L') == moves.count('R') and moves.count('U') == moves.count('D'):
            return True
        else:
            return False


if __name__ == "__main__":
    slt = Solution()
    res = slt.judgeCircle("UD")
    # res = slt.judgeCircle1("LLL")
    print(res)