LeetCode-python 374.猜数字大小

2019-10-11  本文已影响0人  wzNote

题目链接
难度:简单       类型: 二分查找


我们正在玩一个猜数字游戏。 游戏规则如下:
我从 1 到 n 选择一个数字。 你需要猜我选择了哪个数字。
每次你猜错了,我会告诉你这个数字是大了还是小了。
你调用一个预先定义好的接口 guess(int num),它会返回 3 个可能的结果(-1,1 或 0):

-1 : 我的数字比较小
1 : 我的数字比较大
0 : 恭喜!你猜对了!

示例

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

解题思路


经典的二分法
初始化时,搜索区间为(l, r), 其中,l = 0, r = n,每次取mid=(l+r)//2,根据mid和目标数的大小关系修改猜测的区间:
若mid>目标数,下次的搜索区间为(l, mid-1)
若mid<目标数,下次的搜索区间为(mid+1, r)
若mid = 目标数,直接返回mid

代码实现

# 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
        """       
        l, r = 0, n
        while True:
            mid = (l+r)//2
            res =  guess(mid)
            if  res == 0:
                return mid
            elif res == -1:
                r = mid-1              
            elif res == 1:
                l = mid+1

本文链接:https://www.jianshu.com/p/c195829b362b

上一篇下一篇

猜你喜欢

热点阅读