1. 程式人生 > >Leetcode:874. 模擬行走機器人

Leetcode:874. 模擬行走機器人

如果機器人試圖走到障礙物上方,那麼它將停留在障礙物的前一個網格方塊上,但仍然可以繼續該路線的其餘部分。

返回從原點到機器人的最大歐式距離的平方

 

示例 1:

輸入: commands = [4,-1,3], obstacles = []
輸出: 25
解釋: 機器人將會到達 (3, 4)

示例 2:

輸入: commands = [4,-1,4,-2,4], obstacles = [[2,4]]
輸出: 65
解釋: 機器人在左轉走到 (1, 8) 之前將被困在 (1, 4) 處

 

提示:

  1. 0 <= commands.length <= 10000
  2. 0 <= obstacles.length <= 10000
  3. -30000 <= obstacle[i][0] <= 30000
  4. -30000 <= obstacle[i][1] <= 30000
  5. 答案保證小於 2 ^ 31

解題思路:

座標的hash,模擬題。判斷下一步是否有牆壁,如果沒有則走一步,否則停留在原處,等待下一次方向的改變。

題中,牆壁的範圍是[-30000,30000]的區間,如何判斷一個座標(x,y)是否是牆,可以將(x,y)轉換成long型的整數。

long itoll(int a, int b) {
        return ((a + 30000) << 16) + b + 30000;
    }

注意:一定要將a,b轉換成無符號型,即全部相加30000,否則不是嚴格的一一對應。

C++
static int x = []() {
    std::ios::sync_with_stdio(false);
    cin.tie(NULL);
    return 0;
}();
class Solution {
public:
    long itoll(int a, int b) {
        return ((a + 30000) << 16) + b + 30000;
    }
    int robotSim(vector<int>& commands, vector<vector<int>>& obstacles) {
        char direction[4] = { 'U','R','D','L' };
        unordered_set<long> S;
        int dx[4] = { 0,1,0,-1 };
        int dy[4] = { 1,0,-1,0 };
        int pos = 0;
        int x = 0, y = 0;
        int _max = 0;
        int size = commands.size(), i, j;
        for (i = 1; i <= obstacles.size(); i++) {
            S.insert(itoll(obstacles[i - 1][0],obstacles[i - 1][1]));
        }
        for (i = 1; i <= size; i++) {
            if (commands[i - 1] == -2)  pos = (pos + 3) % 4;
            else if (commands[i - 1] == -1) pos = (pos + 1) % 4;
            else {
                for (j = 1; j <= commands[i - 1]; j++) {
                    if (S.count(itoll(x + dx[pos],y + dy[pos])) != 0) {
                        break;
                    }
                    x = x + dx[pos];
                    y = y + dy[pos];
                }
                _max = max(_max , (x*x + y*y));
            }
        }
        return _max;
    }
};