题目 :猜数字游戏的规定如下:
- 每轮游戏,我都会从 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))
}
提交截图 :