lintcode 1493. 模拟行走机器人

机器人在一个无限大小的网格上行走,从点 (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

思路:记录每一次转身所面对的方向在x,y方向上的可移动距离,接着,在机器人现所在的点每次移动一次,看移动后的位置是否存在障碍物,若存在,则不前进,否则前进,到最后点,计算出距离的平方。

class Solution {
public:
    /**
     * @param commands: type: List[int]
     * @param obstacles: type: List[List[int]]
     * @return: Return the square of the maximum Euclidean distance
     */
    int robotSim(vector<int> &commands, vector<vector<int>> &obstacles) {
        // write your code here
        vector<vector<int>> dir={{0,1},{-1,0},{0,-1},{1,0}};
        vector<int> direction=dir[0];
        int point[2]={0,0};
        int index=0;
        for (int i = 0; i < commands.size(); i++) {
            /* code */
            if(commands[i]==-1){direction=dir[abs(index-1+4)%4];index=abs(index-1+4)%4;}
            else if(commands[i]==-2) {direction=dir[abs(index+1)%4];index=abs(index+1)%4;}
            else 
            {
                int step=commands[i];
                while(step--)
                {
                    vector<int> tmp = { point[0] + direction[0], point[1] + direction[1] };
                    if(!count(obstacles.begin(),obstacles.end(),tmp))
                    {
                        point[0]+=direction[0];
                        point[1]+=direction[1];
                    }
                    else break;
                }
            }
        }
        return pow(point[0],2)+pow(point[1],2);
    }
};
发布了369 篇原创文章 · 获赞 15 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43981315/article/details/103991204