[Golang]力扣Leetcode - 374. 猜数字大小（二分查找）

[Golang]力扣Leetcode - 374. 猜数字大小（二分查找）

题目：猜数字游戏的规则如下：

• 每轮游戏，我都会从 1 到 n 随机选择一个数字。 请你猜选出的是哪个数字。
• 如果你猜错了，我会告诉你，你猜测的数字比我选出的数字是大了还是小了。

你可以通过调用一个预先定义好的接口 int guess(int num) 来获取猜测结果，返回值一共有 3 种可能的情况（-1，1 或 0）：

• -1：我选出的数字比你猜的数字小 pick < num
• 1：我选出的数字比你猜的数字大 pick > num
• 0：我选出的数字和你猜的数字一样。恭喜！你猜对了！pick == num
  返回我选出的数字。

链接： 力扣Leetcode - 374. 猜数字大小.

示例 1：

示例 2：

示例 3：

示例 4：

思路：用二分查找，mid = left + right

• 如果 guess(mid) = -1，证明我们猜的数比他想的大，right = mid - 1 缩小范围；
• 如果 guess(mid) = 1，证明我们猜的数比他想的小，left = mid + 1 缩小范围；
• 如果 guess(mid) = 0，就证明他想的数就是 mid

直至找到他要的数 return 出来

Go代码：

package main

import "fmt"

func guess(num int) int {
	// 1~10 假设我想的数是 6
	if num < 6 {
		return 1
		// 1：我选出的数字比你猜的数字大
	} else if num > 6 {
		return -1
		// -1：我选出的数字比你猜的数字小
	} else {
		return 0
	}
}

func guessNumber(n int) int {
	// 二分查找
	left, right := 1, n
	for left <= right {
		mid := (left + right) / 2
		guess := guess(mid)
		if guess == -1 {
			right = mid - 1
		} else if guess == 1 {
			left = mid + 1
		} else if guess == 0 {
			return mid
		}
	}
	return left
}

func main() {
	fmt.Println(guessNumber(10))
}

提交截图：