一个从小到大的有序列表,用最优方式查找目标元素的下标,如果没有,就返回下标适合插入的地方
如:ls=[1,3,5] 查找 2,如果没有找到,就返回2可以插入地方的下标,返回1
def find(ls,tag):
if ls[0]>tag:
return 0
if ls[-1]<tag:
return len(ls)
first=0
last=len(ls)-1
while first<=last:
if first == last:
if ls[first]<tag:
return first+1
elif ls[first]>tag:
return first
mid=(first+last)//2
if ls[mid]<tag:
first=mid+1
elif ls[mid]>tag:
last=mid-1
elif ls[mid]==tag:
return mid
if __name__ == "__main__":
ls=[1,3,5]
print(find(ls,2))
二:如果是单纯的查元素更简单
def find(ls,tag):
first=0
last=len(ls)
while first<=last:
mid=(first+last)//2
if ls[mid]>tag:
last=mid-1
if ls[mid]<tag:
first=mid+1
if ls[mid]==tag:
print("找到了")
return 0
print("没找到")