前言:剑指offer刷题系列
问题:
猜数字游戏的规则如下:
- 每轮游戏,我都会从 1 到 n 随机选择一个数字。 请你猜选出的是哪个数字。
- 如果你猜错了,我会告诉你,你猜测的数字比我选出的数字是大了还是小了。
你可以通过调用一个预先定义好的接口 int guess(int num)
来获取猜测结果,返回值一共有 3 种可能的情况(-1
,1
或 0
):
- -1:我选出的数字比你猜的数字小
pick < num
- 1:我选出的数字比你猜的数字大
pick > num
- 0:我选出的数字和你猜的数字一样。恭喜!你猜对了!
pick == num
返回我选出的数字。
示例:
输入:n = 10, pick = 6
输出:6
思路:
非常简单的一道题,好像是我步入计算机专业,老师教我们的第一个编码游戏,今天正好写到这个题,就想起那段时光,真好,不像现在每天都被我的论文支配着😫。
在我的方法中,定义了两个变量left
和right
,分别表示猜测范围的左边界和右边界。初始时,将left
设为1,将right
设为n
。
接下来进入一个循环,条件是left <= right
。在每次循环中,计算中间位置mid
,然后调用guess(mid)
函数进行猜测。
如果guess(mid)
函数返回0,表示猜测成功,直接返回mid
作为结果。
如果guess(mid)
函数返回-1,表示猜测失败,说明目标数字比mid
要小,因此将right
更新为mid - 1
,缩小猜测范围。
如果guess(mid)
函数返回其他值,表示猜测的结果介于0和-1之间,即目标数字比mid
要大,因此将left
更新为mid + 1
,缩小猜测范围。
通过不断缩小猜测范围,最终找到目标数字并返回结果。
总结来说,这段代码实现了一个二分查找算法,用于猜测一个数字。
基于上述思考,代码如下:
class Solution:
def guessNumber(self, n: int) -> int:
left = 1
right = n
while left <= right:
mid = (left + right) // 2
ret = guess(mid)
if ret == 0:
return mid
elif ret == -1:
right = mid - 1
else:
left = mid + 1