LeetCode刷題記錄——第374題(猜數字大小)
阿新 • • 發佈:2019-01-13
題目描述
我們正在玩一個猜數字遊戲。 遊戲規則如下:
我從 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
high = n
while low <= high:
mid = (low + high) // 2
if guess(mid) == -1:
high = mid - 1
elif guess(mid) == 1:
low = mid + 1
else:
return mid