来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/guess-number-higher-or-lower
我们正在玩一个猜数字游戏。 游戏规则如下:
我从 1 到 n 选择一个数字。 你需要猜我选择了哪个数字。
每次你猜错了,我会告诉你这个数字是大了还是小了。
你调用一个预先定义好的接口 guess(int num)
,它会返回 3 个可能的结果(-1
,1
或 0
):
-1 : 我的数字比较小
1 : 我的数字比较大
0 : 恭喜!你猜对了!
示例 :
输入: n = 10, pick = 6
输出: 6
二分查找的扩展,隶属于二分查找吧
# The guess API is already defined for you.
# @param num, your guess
# @return -1 if my number is lower, 1 if my number is higher, otherwise return 0
# def guess(num):
class Solution(object):
def guessNumber(self, n):
"""
:type n: int
:rtype: int
"""
minc, maxc = 0, n-1
while minc <= maxc:
mid = (minc+maxc)//2
if guess(mid)==-1:
maxc = mid-1
if guess(mid)==1:
minc = mid+1
if guess(mid)==0:
return mid
return mid+1
执行用时 :24 ms, 在所有 Python 提交中击败了56.95%的用户
内存消耗 :11.7 MB, 在所有 Python 提交中击败了36.89%的用户