力扣:374. 猜数字大小

306 阅读2分钟

前言:剑指offer刷题系列

问题:

猜数字游戏的规则如下:

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

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

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

返回我选出的数字。

示例:

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

思路:

非常简单的一道题,好像是我步入计算机专业,老师教我们的第一个编码游戏,今天正好写到这个题,就想起那段时光,真好,不像现在每天都被我的论文支配着😫。

在我的方法中,定义了两个变量leftright,分别表示猜测范围的左边界和右边界。初始时,将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

执行结果如下图:

image-20230920220244215.png