【Python】 猜數字大小 #二分查詢
阿新 • • 發佈:2018-11-21
我們正在玩一個猜數字遊戲。 遊戲規則如下:
我從 1 到 n 選擇一個數字。 你需要猜我選擇了哪個數字。
每次你猜錯了,我會告訴你這個數字是大了還是小了。
你呼叫一個預先定義好的介面 guess(int num),它會返回 3 個可能的結果(-1,1 或 0):
-1 : 我的數字比較小
1 : 我的數字比較大
0 : 恭喜!你猜對了!
示例 :
輸入: n = 10, pick = 6
輸出: 6
解法一:
條件設定pick是隱藏的,所以無法mid==pick來判斷,需guess()函式的三個返回值來定位。
這道題當然可以用從1到n的遍歷方法。但二分查詢更高效的
class Solution(object): def guessNumber(self, n): import math lo,hi=1,n while(lo<=hi): mid=math.ceil((lo+hi)/2) pick=.guess(mid) # pick=Solution2().guess(mid) #本地測試 if(pick==0): return int(mid) elif(pick==1): lo=mid+1 else: hi=mid-1
guess()函式,用於本地測試的
class Solution2(object):
def guess(self,num):
n=6
if(num>n):
return 1
elif(num<n):
return -1
else:
return 0
x=Solution().guessNumber(10)
print(x)
25 / 25 個通過測試用例
執行用時:28 ms