LeetCode-Python-374. 猜数字大小

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

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

示例 :

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

思路:

二分查找。注意guess返回的结果,如果是-1,说明猜大了,下一次应该猜更小的数字,

                                                         如果是1,说明猜小了,下一次应该猜更大的数字。

# 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
        """
        lo, hi = 1, n
        while(lo <= hi):
            mid = lo + (hi - lo) / 2
            val = guess(mid)
            if val == 0:
                return mid
            elif val == 1:
                lo = mid + 1
            else:                
                hi = mid - 1
                

 

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/87923378