1. 程式人生 > >【Python】 猜數字大小 #二分查詢

【Python】 猜數字大小 #二分查詢

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