leetcode657機器人能否返回原點

1.問題描述


在二維平面上,有一個機器人從原點 (0, 0) 開始。給出它的移動順序,判斷這個機器人在完成移動後是否在 (0, 0) 處結束。

移動順序由字符串表示。字符 move[i] 表示其第 i 次移動。機器人的有效動作有 R(右),L(左),U(上)和 D(下)。如果機器人在完成所有動作後返回原點,則返回 true。否則,返回 false。

注意:機器人“面朝”的方向無關緊要。 “R” 將始終使機器人向右移動一次,“L” 將始終向左移動等。此外,假設每次移動機器人的移動幅度相同。

 

示例 1:

輸入: "UD"
輸出: true
解釋:機器人向上移動一次,然後向下移動一次。所有動作都具有相同的幅度,因此它最終回到它開始的原點。因此,我們返回 true。
示例 2:

輸入: "LL"
輸出: false
解釋:機器人向左移動兩次。它最終位於原點的左側,距原點有兩次 “移動” 的距離。我們返回 false,因爲它在移動結束時沒有返回原點。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/robot-return-to-origin

2.python實現


  • 1.我的思路

分別對RL和UD統計計數,R+1,L-1,U+1,D-1如果在X方向和y方向相加和都等於0,那麼回到原點

class Solution(object):
    def judgeCircle(self, moves):
        """
        :type moves: str
        :rtype: bool
        """
        sum_x = 0
        sum_y = 0
        dict_x = {'R':1,'L':-1,'U':0,'D':0}
        dict_y = {'R':0,'L':0,'U':1,'D':-1}
        
        for c in moves:
            sum_x += dict_x[c]
            sum_y += dict_y[c]
        #if sum_x==0 and sum_y==0:
            #return True
        #else:
            #return False
        return sum_x==0 and sum_y==0
  • 2.排名前幾的思路:直接調用list的內置函數count(object).
class Solution(object):
    def judgeCircle(self, moves):
        """
        :type moves: str
        :rtype: bool
        """
        return moves.count('R')==moves.count('L') and moves.count('U')==moves.count('D')

3.寫代碼過程遇到的問題


1.pythonic,更簡潔

 #if sum_x==0 and sum_y==0:
     #return True
 #else:
     #return False

寫爲下面這一行,更簡潔
 return sum_x==0 and sum_y==0

2.list的內置函數list.count(object),用於統計某個元素在列表中出現的次數。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章