题目
874. 模拟行走机器人文章来源地址https://www.toymoban.com/news/detail-642858.html
题解思路
- 初始方向朝y轴正方向,遇到指令command == -1 则向右转, 若为 -2 则向左转
- 定义方向[-1,0]、[0,1]、[1,0]、[0,-1] 分别为朝x轴负方向, y轴正方向, x轴正方向,y轴负方向
- 初始方向为[0,1], 若向右转 则方向变为[-1,0]、若向左转方向变为[1,0]。
- 若向右转则不断 向右递加, 向左转则向左递减
- 同时建立集合set 存储有障碍的点。(set集合查询时间复杂度为o(1))
代码
C++
class Solution {
public:
int robotSim(vector<int>& commands, vector<vector<int>>& obstacles) {
int dirs[4][2] = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
int sx = 0, sy = 0, res = 0, d = 1;
set<pair<int, int>> mp;
for(int i = 0; i < obstacles.size(); ++i){
pair<int, int> t(obstacles[i][0], obstacles[i][1]);
mp.insert(t);
}
for (int c : commands){
if (c < 0){
d += c == -1 ? 1 : -1;
d %= 4;
if (d < 0){
d += 4;
}
}else{
for (int i = 0; i < c; ++i){
int nx = sx + dirs[d][0];
int ny = sy + dirs[d][1];
pair<int, int> t(nx, ny);
if (mp.count(t)){
break;
}
res = max(res, nx * nx + ny * ny);
sx = nx;
sy = ny;
}
}
}
return res;
}
};
Python
class Solution:
def robotSim(self, commands: List[int], obstacles: List[List[int]]) -> int:
dirs = [[-1, 0], [0, 1], [1, 0], [0, -1]]
sx, sy = 0, 0
d = 1
res = 0
mp = set([tuple(i) for i in obstacles])
for c in commands:
if c < 0:
d += 1 if c == -1 else -1
d %= 4
else:
for i in range(c):
if tuple([sx + dirs[d][0], sy + dirs[d][1]]) in mp:
break
else:
sx += dirs[d][0]
sy += dirs[d][1]
res = max(res, sx*sx + sy * sy)
return res
文章来源:https://www.toymoban.com/news/detail-642858.html
到了这里,关于每日一题(set集合)-874. 模拟行走机器人的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!