拟人算法系列文章,以容易理解的方式描述算法,点赞收藏不迷路
0~n 中缺失的数字
def get_missing_num(a: list[int]) -> int:
lo = 0
hi = len(a) - 1
while lo <= hi:
mid = (lo + hi) // 2
if a[mid] != mid:
if mid == 0 or a[mid - 1] == mid - 1:
return mid
hi = mid - 1
else:
lo = mid + 1
if lo == len(a):
return len(a)
return -1
if __name__ == '__main__':
a = [1, 2, 4, 5]
print(get_missing_num(a)) # 3
采用二分查找的思路,只是这个人现在可以根据自己所在位置的值是否等于目标值,调整小弟让范围缩小
- 如果等于说明在右边,于是调整第一个小弟的位置
- 如果不等于说明在左边,于是调整第二个小弟的位置
同时,在调整任何一个小弟时,可以额外判断下,前一位或者后一位是不是和自己的值不同,如果不同则找到了