皮卡丘python写leetcode——Day5

·657 机器人是否可以返回原点

在二维平面上，有一个机器人从原点 (0, 0) 开始。给出它的移动顺序，判断这个机器人在完成移动后是否在 (0, 0) 处结束。

移动顺序由字符串 moves 表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R（右），L（左），U（上）和 D（下）。

如果机器人在完成所有动作后返回原点，则返回 true。否则，返回 false。

注意：机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次，“L” 将始终向左移动等。此外，假设每次移动机器人的移动幅度相同。

·解题思路

只要向上和向下的次数、向左和向右的次数均相等，那么机器人就可以返回原点

·解题代码

class Solution(object):
    def judgeCircle(self, moves):

        U_num = 0
        D_num = 0
        R_num = 0
        L_num = 0
        for i in moves:
            if i == 'U':
                U_num += 1
            elif i == 'D':
                D_num += 1
            elif i == 'R':
                R_num += 1
            else:
                L_num += 1
        return U_num == D_num and R_num == L_num