leetcode 1041——困於環中的機器人

描述:python

在無限的平面上,機器人最初位於 (0, 0) 處,面朝北方。機器人能夠接受下列三條指令之一:c++

  • "G":直走 1 個單位
  • "L":左轉 90 度
  • "R":右轉 90 度

機器人按順序執行指令 instructions,並一直重複它們。spa

只有在平面中存在環使得機器人永遠沒法離開時,返回 true。不然,返回 falsecode

示例 1:blog

輸入:"GGLLGG"
輸出:true
解釋:
機器人從 (0,0) 移動到 (0,2),轉 180 度,而後回到 (0,0)。
重複這些指令,機器人將保持在以原點爲中心,2 爲半徑的環中進行移動。

示例 2:string

輸入:"GG"
輸出:false
解釋:
機器人無限向北移動。

示例 3:io

輸入:"GL"
輸出:true
解釋:
機器人按 (0, 0) -> (0, 1) -> (-1, 1) -> (-1, 0) -> (0, 0) -> ... 進行移動。

思路:循環四次後能回到原點(0,0)的符合
C++:
class Solution {
public:
    bool isRobotBounded(string instructions) {
        int a=0,b=0,c=0;//(a,b)表示座標,c表示方向
        instructions+=instructions+instructions+instructions;
        for(int i=0;i<instructions.length();i++){
            if(instructions[i]=='G'){
                if(c==0)  b++;
                else if(c==1)  a--;
                else if(c==2)  b--;
                else a++;
            }else if(instructions[i]=='L'){
                if(c==3)  c=0;
                else c++;
            }else if(instructions[i]=='R'){
                if(c==0)  c=3;
                else c--;
            }
        }
        return a==0&&b==0;
    }
};
python:
class Solution:
    def isRobotBounded(self, instructions: str) -> bool:
        a=0
        b=0
        c=0
        instructions+=instructions+instructions+instructions
        for i in instructions:
            if i=='G':
                if c==0: b+=1
                elif c==1: a-=1
                elif c==2: b-=1
                else: a+=1
            elif i=='L':
                if c==3: c=0
                else: c+=1
            elif i=='R':
                if c==0: c=3
                else: c-=1
        return a==0 and b==0
相關文章
相關標籤/搜索