374. 猜数字大小【二分查找】

374. 猜数字大小

374. 猜数字大小

我们正在玩猜数字游戏。猜数字游戏的规则如下：

我会从 1 到n随机选择一个数字。 请你猜选出的是哪个数字。

如果你猜错了，我会告诉你，我选出的数字比你猜测的数字大了还是小了。

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

• -1：你猜的数字比我选出的数字大 （即 num > pick）。
• 1：你猜的数字比我选出的数字小 （即 num < pick）。
• 0：你猜的数字与我选出的数字相等。（即 num == pick）。

返回我选出的数字。

示例 1：

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

示例 2：

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

示例 3：

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

提示：

• 1 <= n <= 2^31 - 1
• 1 <= pick <= n

解题思路

Go代码

/** 
 * Forward declaration of guess API.
 * @param  num   your guess
 * @return 	     -1 if num is higher than the picked number
 *			      1 if num is lower than the picked number
 *               otherwise return 0
 * func guess(num int) int;
 */

func guessNumber(n int) int {
    // 题目说明了是1~n中的某个数
    l,r := 1,n
    for l < r {
        mid := (l + r) / 2
        fmt.Println(mid)
        if guess(mid) == -1 {
            r = mid - 1
        } else if guess(mid) == 1 {
            l = mid + 1
        } else {
            return mid
        }
    }

    // 退出上面的for时，l最后是和r相等的，也就是最后肯定是只剩下一个数没猜了，这个数肯定是答案
    return l
}