374. 猜数字大小
2019-10-23 本文已影响0人
zhangsini
题目
我们正在玩一个猜数字游戏。 游戏规则如下:
我从 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
"""
low = 1
hei = n
while low <= hei:
mid = (hei + low) // 2 # 获取中间值
ret = guess(mid)
if ret == 0:
return mid
elif ret == 1: # 猜小了 低的往上移
low = mid + 1
else: # 猜大了 高的往下移
hei = mid - 1