[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：

输入：n = 10, pick = 6
输出：6

示例 2：

输入：n = 1, pick = 1
输出：1

示例 3：

输入：n = 2, pick = 1
输出：1

示例 4：

输入：n = 2, pick = 2
输出：2

思路：用二分查找，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))
}

提交截图：