刷题日记 374. 猜数字大小

40 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第10天,点击查看活动详情

一、题目描述:

374. 猜数字大小 - 力扣(LeetCode) (leetcode-cn.com)

猜数字游戏的规则如下:

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

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

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

返回我选出的数字。

示例 1:

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

示例 2:

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

示例 3:

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

示例 4:

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

提示:

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

二、思路分析:

思路就是简单的二分查找,主要是练习写法。

首先这个target是在1-n中一定可以确认的,因此使用while left <= right 和 while left < right作为循环条件都可以(其实还可以用while guess(mid)作为循环条件)

如果使用 while left <= right,就是最基础的二分查找写法,而且不用考虑循环结束后的情况(因为不可能发生)

如果使用 while left < right,就要考虑到循环结束的情况,此时left和right指向同一个元素,该元素返回就行,另外因为n的最小值为1,也不用考虑返回left还是right,都一样;还有就是mid要用另一种写法让它“倾向于right”

如果使用 while guess(mid),就和基础的一样,条件判断上改一下就是了。

三、AC 代码:

# while guess(mid)写法
class Solution:
    def guessNumber(self, n: int) -> int:
        i, j = 1, n
        mid = (i+j)//2
        while guess(mid):
            if guess(mid) < 0:
                j = mid - 1
            else:
                i = mid + 1
            mid = (i+j)//2
        return mid