题意:
力扣团队买了一个可编程机器人,机器人初始位置在原点(0, 0)。
小伙伴事先给机器人输入一串指令command,机器人就会无限循环这条指令的步骤进行移动。指令有两种:
U: 向y轴正方向移动一格
R: 向x轴正方向移动一格。
不幸的是,在 xy 平面上还有一些障碍物,他们的坐标用obstacles表示。机器人一旦碰到障碍物就会被损毁。
给定终点坐标(x, y),返回机器人能否完好地到达终点。如果能,返回true;否则返回false。
数据范围:
2 <= command的长度 <= 1000
command由U,R构成,且至少有一个U,至少有一个R
0 <= x <= 1e9, 0 <= y <= 1e9
0 <= obstacles的长度 <= 1000
obstacles[i]不为原点或者终点
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/programmable-robot
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法:
设su[i]为移动序列前i步'U'的数量,sr[i]为移动序列前i步'R'的数量.
设移动序列的长度为n,
如果机器人能够到达(x,y),
那么一定有su[n]*k+su[i]=x且su[n]*k+sr[i]=y,
其中k是满足su[n]*k<=x&&su[n]*k<=y的最大值,
k可以O(1)计算,然后O(n)从[0,n]枚举i,
判断su[n]*k+su[i]=x且su[n]*k+sr[i]=y是否满足即可判断能否到达(x,y).
对于这题,就是判断起点能否到达终点(x,y),
且不能到达(0,0)到(x,y)矩形中的任何障碍点.
code:
class Solution {
public:
static const int maxm=1e3+5;
int su[maxm],sr[maxm];
int n;
bool check(int x,int y){
int k=min(x/sr[n],y/su[n]);
int xx=x-k*sr[n],yy=y-k*su[n];
for(int i=0;i<=n;i++){
if(sr[i]==xx&&su[i]==yy){
return 1;
}
}
return 0;
}
bool robot(string s, vector<vector<int>>& o, int x, int y) {
memset(su,0,sizeof su);
memset(sr,0,sizeof sr);
n=s.size();
s='p'+s;
for(int i=1;i<=n;i++){
su[i]=su[i-1]+(s[i]=='U');
sr[i]=sr[i-1]+(s[i]=='R');
}
for(auto i:o){
if(i[0]>x||i[1]>y)continue;
if(check(i[0],i[1])){
return 0;
}
}
if(check(x,y)){
return 1;
}
return 0;
}
};