亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

從有序序列中求最大 不大于目標 的數的下標的二分查找怎么寫比較優雅?

從有序序列中求最大 不大于目標 的數的下標的二分查找怎么寫比較優雅?

幕布斯6054654 2019-04-19 16:13:09
本身這個題挺簡單的,但是如果增加這樣一個要求怎么寫:如果序列中有多個等于目標的數,則可以傳入一個flag參數,來決定返回等于目標的數最大下標還是最小下標舉個例子:binsearch([1,3,3],3,"max")返回2binsearch([1,3,3],3,"min")返回1binsearch([1,1],2,"max")和binsearch([1,1],2,"min")都返回1就是只有當數列中有多個等于目標的數時flag參數才有意義。我寫的只考慮flag=="min"的代碼,我覺得最后幾行if已經很不優雅了,flag=="max"怎么也寫不對defbinsearch(a,target,flag="min"):'''Findtheindexofthemaxonenotgreaterthantarget'''l,r=0,len(a)-1while(l=target:r=midelse:l=mid+1ifa[l]==target:returnlifa[r]==target:returnrifa[r]
查看完整描述

2 回答

?
慕森王

TA貢獻1777條經驗 獲得超3個贊

我后來又想了想,同時參考了樓上@brayden的思想,把兩種功能的實現完全分開,重新寫了一個
defbinsearchmax(a,target):
l,r=0,len(a)-1
while(lmid=l+(r-l)/2
ifa[mid]<=target:
l=mid
else:
r=mid-1
ifa[r]<=target:returnr
ifa[l]<=target:returnl
return-1
defbinsearchmin(a,target):
l,r=0,len(a)-1
while(lmid=l+(r-l)/2
ifa[mid]l=mid
else:
r=mid
ifa[l]==target:returnl
ifa[r]==target:returnr
ifa[r]ifa[l]return-1
defbinsearch(a,target,flag="min"):
ifflag=="min":
returnbinsearchmin(a,target)
else:
returnbinsearchmax(a,target)
assert(binsearch([1,3],1)==0)
assert(binsearch([1,3],3)==1)
assert(binsearch([1,3],2)==0)
assert(binsearch([1,1],2)==1)
assert(binsearch([1,1,3],1)==0)
assert(binsearch([2,2],1)==-1)
assert(binsearch([1,3,5],1)==0)
assert(binsearch([1,3,5],5)==2)
assert(binsearch([1,3,5,5],5)==2)
assert(binsearch([1,1,1,2,2,2,3,4,5,5],2)==3)
assert(binsearch([1,3],1,"max")==0)
assert(binsearch([1,3],3,"max")==1)
assert(binsearch([1,3],2,"max")==0)
assert(binsearch([1,1],2,"max")==1)
assert(binsearch([1,1,3],1,"max")==1)
assert(binsearch([2,2],1,"max")==-1)
assert(binsearch([1,3,5],1,"max")==0)
assert(binsearch([1,3,5],5,"max")==2)
assert(binsearch([1,3,5,5],5,"max")==3)
我自己構造的測試數據全部通過了。總結一下,我覺得我一開始試圖把兩種功能的邏輯混到一起的這個思想是導致我寫不出來的一個重要原因。再次感謝樓上的@brayden
                            
查看完整回答
反對 回復 2019-04-19
  • 2 回答
  • 0 關注
  • 1029 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號