题目描述:
机器人在一个无限大小的 XY 网格平面上行走,从点 (0, 0) 处开始出发,面向北方。该机器人可以接收以下三种类型的命令 commands :
-2 :向左转 90 度
-1 :向右转 90 度
1 <= x <= 9 :向前移动 x 个单位长度
在网格上有一些格子被视为障碍物 obstacles 。第 i 个障碍物位于网格点 obstacles[i] = (xi, yi) 。
机器人无法走到障碍物上,它将会停留在障碍物的前一个网格方块上,但仍然可以继续尝试进行该路线的其余部分。
返回从原点到机器人所有经过的路径点(坐标为整数)的最大欧式距离的平方。(即,如果距离为 5 ,则返回 25 )
示例:
输入:commands = [4,-1,3], obstacles = []
输出:25
解释:
机器人开始位于 (0, 0):
1. 向北移动 4 个单位,到达 (0, 4)
2. 右转
3. 向东移动 3 个单位,到达 (3, 4)
距离原点最远的是 (3, 4) ,距离为 32 + 42 = 25
算法思想:
按照题意在XY数组上移动,碰到障碍物就暂停移动,直接执行下一条命令。使用一个集合维护障碍物的位置,每次移动时,检查下一步是否是障碍物。
使用方向数组dx = {0,1,0,-1},dy = {1,0,-1,0}表示朝一个方向移动一个位置的变化量,使用 d 的值来索引方向数组,d的值为{0{北},1{东},2{南},3{西}}(与方向数组转向保持一致)向东对应的方向数组索引是 dx[1],dy[1], 表现在d 的变化上就是 d =(d + 1)%4, 由于西是最后一个,类似python计法向西对应的方向数组的索引是 -1, 为了防止 d < 0, 所以加4对应 d 的变化就是 d = (d - 1 + 4) % 4 .
c++代码实现:
class Solution {
public:
int robotSim(vector<int>& commands, vector<vector<int>>& obstacles) {
for(auto obstacle : obstacles){
string temp = getStr(obstacle);
hash.insert(temp);
}
for(auto iter = hash.begin(); iter != hash.end(); ++iter)
cout << *iter << endl;
int d = 0;
int dx[4] = {0,1,0,-1}, dy[4] = {1,0,-1,0};
int x = 0, y = 0;
int ans = 0;
for(auto command : commands){
if(command == -2) d = (d + 3) % 4;
else if(command == -1) d = (d + 1) % 4;
else{
// 一步一步走,如果走到(x,y)处发现障碍物了,退回障碍物前一个地方。
for(int i = 0; i < command; i++){
int a = x + dx[d];
int b = y + dy[d];
if(hash.count(getStr({a,b}))) break;
x = a, y = b;
// ans 等于离原点最远的点的距离
ans = max(ans,(x * x + y * y));
}
}
}
return ans;
}
private:
unordered_set<string> hash;
string getStr(vector<int> obstacle){
return to_string(obstacle[0]) + "#" + to_string(obstacle[1]);
}
};
时间复杂度:
时间复杂度: O(N + K), N, K 是 commands与obstacles的长度。