机器人能否返回原点
在二维平面上,有一个机器人从原点 (0, 0) 开始。给出它的移动顺序,判断这个机器人在完成移动后是否在 (0, 0) 处结束。
移动顺序由字符串 moves 表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R(右),L(左),U(上)和 D(下)。
如果机器人在完成所有动作后返回原点,则返回 true。否则,返回 false。
注意:机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次,“L” 将始终向左移动等。此外,假设每次移动机器人的移动幅度相同。
示例1:
输入: moves = "UD"
输出: true
解释:机器人向上移动一次,然后向下移动一次。所有动作都具有相同的幅度,因此它最终回到它开始的原点。因此,我们返回 true。
示例2:
输入: moves = "LL"
输出: false
解释:机器人向左移动两次。它最终位于原点的左侧,距原点有两次 “移动” 的距离。我们返回 false,因为它在移动结束时没有返回原点。
提示:
- 1 <= moves.length <= 2 * 104
- moves 只包含字符 'U', 'D', 'L' 和 'R'
思路:
使用一个二维数组dir表示移动方向,然后使用switch将字符串映射为dir每行的索引。
int map(char ch){
switch(ch){
case 'U':
return 0;
case 'R':
return 1;
case 'D':
return 2;
case 'L':
return 3;
default:
return 0;
}
}
bool judgeCircle(char* moves) {
int dir[][2] = {{1,0},{0,1},{-1,0},{0,-1}};
int row = 0,col = 0;
for(int i=0;moves[i]!='\0';i++){
int idx = map(moves[i]);
row+=dir[idx][0];
col+=dir[idx][1];
}
if(row!=0||col!=0){
return false;
}
return true;
}