携手创作,共同成长!这是我参与「掘金日新计划 · 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