LeetCode | 困于环中的机器人

一、题目描述

在无限的平面上，机器人最初位于 (0, 0) 处，面朝北方。机器人可以接受下列三条指令之一：

“G”：直走 1 个单位
“L”：左转 90 度
“R”：右转 90 度
机器人按顺序执行指令 instructions，并一直重复它们。

只有在平面中存在环使得机器人永远无法离开时，返回 true。否则，返回 false。

示例 1：

输入："GGLLGG"
输出：true
解释：
机器人从 (0,0) 移动到 (0,2)，转 180 度，然后回到 (0,0)。
重复这些指令，机器人将保持在以原点为中心，2 为半径的环中进行移动。

示例 2：

输入："GG"
输出：false
解释：
机器人无限向北移动。

示例 3：

输入："GL"
输出：true
解释：
机器人按 (0, 0) -> (0, 1) -> (-1, 1) -> (-1, 0) -> (0, 0) -> ... 进行移动。

提示：

1 <= instructions.length <= 100
instructions[i] 在 {'G', 'L', 'R'} 中

二、题解思路

定义出方向和机器人在各个方向上位移的距离。只要判断在两个相反的方向上位移的距离是一样的就说明机器人回到了原点。

三、程序实例

class Solution {
public:
    bool isRobotBounded(string instructions) {
        int direct = 0;
        vector<int> axis{0, 0, 0, 0};
        
        int i = 0;
        while(i < 4) // 4个方向转一轮，回原点
        {
            for(auto ch : instructions)
            {
                switch (ch)
                {
                    case 'G':
                        axis[direct] += 1; break; // 相反方向的位移一样说明回到了原点
                    case 'L':
                        direct = (direct + 3) % 4; break;
                    case 'R':
                        direct = (direct + 1) % 4; break;
                    default:
                        return false;
                }
            }
            if(axis[0] == axis[2] && axis[1] == axis[3]) // 一次就回到原点那么直接返回
                return true;
            i++;
        }
        return (axis[0] == axis[2] && axis[1] == axis[3]) ? true : false;
    }
};