Leetcode:874. 模擬行走機器人
阿新 • • 發佈:2019-01-07
如果機器人試圖走到障礙物上方,那麼它將停留在障礙物的前一個網格方塊上,但仍然可以繼續該路線的其餘部分。
返回從原點到機器人的最大歐式距離的平方。
示例 1:
輸入: commands = [4,-1,3], obstacles = [] 輸出: 25 解釋: 機器人將會到達 (3, 4)
示例 2:
輸入: commands = [4,-1,4,-2,4], obstacles = [[2,4]] 輸出: 65 解釋: 機器人在左轉走到 (1, 8) 之前將被困在 (1, 4) 處
提示:
0 <= commands.length <= 10000
0 <= obstacles.length <= 10000
-30000 <= obstacle[i][0] <= 30000
-30000 <= obstacle[i][1] <= 30000
- 答案保證小於
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,否則不是嚴格的一一對應。
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; } }; |