今天也是来一道简单题,leetcode-657-机器人能否回到原点。这道题比较简单,先来看题目。
在二维平面上,有一个机器人从原点 (0, 0) 开始。给出它的移动顺序,判断这个机器人在完成移动后是否在 (0, 0) 处结束。
移动顺序由字符串表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R(右),L(左),U(上)和 D(下)。如果机器人在完成所有动作后返回原点,则返回 true。否则,返回 false。
注意:机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次,“L” 将始终向左移动等。此外,假设每次移动机器人的移动幅度相同。
示例:
输入: "UD"
输出: true
解释:机器人向上移动一次,然后向下移动一次。所有动作都具有相同的幅度,因此它最终回到它开始的原点。因此,我们返回 true。
大概意思就是输入字符串命令,判断机器人是否回到了原点。因为是二维平面,那岂不是可以用数学上的二维知识x轴和y轴。R就表示在X轴上往右走了一步;L表示在x轴上往左走了一步;U表示在Y轴上往上走了一步;D表示在Y轴上往下走了一步。所以可以用一个变量来维护坐标就好了,最后判断变量是否还是(0, 0)。上代码:
var judgeCircle = function(moves) {
let x = 0, y = 0;
for (let i = 0; i < moves.length; i++) {
switch (moves[i]) {
case "R": {
x++;
break;
}
case "U": {
y++;
break;
}
case "L": {
x--;
break;
}
case "D": {
y--;
break;
}
}
}
return x == 0 && y == 0
};
成功解答了,但有点不够优雅。仅仅是解决这么简单的问题,却要这么长的代码,是不是可以再优化一下。思考再三,其实我们不需要进行加加减减的操作,只需要统计下左和右的数量是否相等;以及上和下的数量是否相等就好了:
var judgeCircle = function(moves) {
if (moves.length % 2 != 0) return false; // 如果字符串长度为奇数,则可以认为不可能回到原点了
let res = {U: 0,D: 0,R: 0,L: 0}
for (let i = 0; i < moves.length; i++) {res[moves[i]] ++}
return res.U === res.D && res.R === res.L
};
这样就简洁了好多,看起来舒服多了。还加多了一条判断,就是如果指令的长度是奇数时,则机器人是不能回到原点的,所以可以直接return false,可以加快判断速度。