1. 程式人生 > >LeetCode刷題記錄——第374題(猜數字大小)

LeetCode刷題記錄——第374題(猜數字大小)

題目描述

我們正在玩一個猜數字遊戲。 遊戲規則如下:
我從 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