[Golang]力扣Leetcode - 657. 机器人能否返回原点（模拟）

[Golang]力扣Leetcode - 657. 机器人能否返回原点（模拟）

题目：在二维平面上，有一个机器人从原点 (0, 0) 开始。给出它的移动顺序，判断这个机器人在完成移动后是否 在 (0, 0) 处结束。

移动顺序由字符串 moves 表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R（右），L（左），U（上）和 D（下）。

如果机器人在完成所有动作后返回原点，则返回 true。否则，返回 false。

注意：机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次，“L” 将始终向左移动等。此外，假设每次移动机器人的移动幅度相同。

链接： 力扣Leetcode - 657. 机器人能否返回原点.

示例1：

输入: moves = “UD”
输出: true
解释：机器人向上移动一次，然后向下移动一次。所有动作都具有相同的幅度，因此它最终回到它开始的原点。因此，我们返回 true。

示例 2:

输入: moves = “LL”
输出: false
解释：机器人向左移动两次。它最终位于原点的左侧，距原点有两次 “移动” 的距离。我们返回 false，因为它在移动结束时没有返回原点。

思路：我们设机器人当前所在的坐标为 (x,y)，起始时 x=0，y=0。接下来我们遍历指令并更新机器人的坐标：

• 如果指令是 U，则令 y=y−1
• 如果指令是 D，则令 y=y+1
• 如果指令是 L，则令 x=x−1
• 如果指令是 R，则令 x=x+1

最后判断 (x,y) 是否为 (0,0) 即可。

go代码如下：

package main
 
import "fmt"
 
func judgeCircle(moves string) bool {
	x, y := 0, 0
	for i := 0; i < len(moves); i++ {
		if moves[i] == 'U' {
			y--
		} else if moves[i] == 'D' {
			y++
		} else if moves[i] == 'L' {
			x--
		} else if moves[i] == 'R' {
			x++
		}
	}
	if x == 0 && y == 0 {
		return true
	}
	return false
}
 
func main() {
	fmt.Println(judgeCircle("UD"))
}
 

提交截图：

或者：

package main
 
import "fmt"
 
func judgeCircle(moves string) bool {
	x, y := 0, 0
	length := len(moves)
	for i := 0; i < length; i++ {
		switch moves[i] {
		case 'U':
			y--
		case 'D':
			y++
		case 'L':
			x--
		case 'R':
			x++
		}
	}
	return x == 0 && y == 0
}
 
func main() {
	fmt.Println(judgeCircle("UD"))
}