LeetCode 874. 模拟行走机器人（set）

Michael阿明

发布于 2020-07-13 16:12:33

3650

发布于 2020-07-13 16:12:33

文章被收录于专栏：Michael阿明学习之路Michael阿明学习之路

1. 题目

机器人在一个无限大小的网格上行走，从点 (0, 0) 处开始出发，面向北方。该机器人可以接收以下三种类型的命令：

-2：向左转 90 度
-1：向右转 90 度
1 <= x <= 9：向前移动 x 个单位长度

在网格上有一些格子被视为障碍物。

第 i 个障碍物位于网格点 (obstacles[i][0], obstacles[i][1])

如果机器人试图走到障碍物上方，那么它将停留在障碍物的前一个网格方块上，但仍然可以继续该路线的其余部分。

返回从原点到机器人的最大欧式距离的平方。

示例 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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/walking-robot-simulation 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

class Solution {
	vector<vector<int>> dir = {{0,1},{1,0},{0,-1},{-1,0}};//方向
public:
	int robotSim(vector<int>& cmd, vector<vector<int>>& ob) {
		int d = 0, x = 0, y = 0, maxDis = 0, i;
		set<pair<int,int>> s;//快速查找，如果用哈希set需要自己写哈希函数
		for(vector<int> o:ob)
			s.insert(make_pair(o[0],o[1]));

		for(int c : cmd) 
		{
			if(c == -2)//左转
				d = (d-1+4)%4;
			else if(c == -1)//右转
				d = (d+1)%4;
			else //模拟
			{
				for(i=0; i<c; i++) 
				{
					x += dir[d][0];
					y += dir[d][1];
					if(s.find(make_pair(x,y)) != s.end()) 
					{	//是障碍物
						x -= dir[d][0];
						y -= dir[d][1];
						break;//退回一格，停止
					}
				}
				maxDis = max(maxDis, x*x+y*y);
			}
		}
		return maxDis;
    }
};