python

怎么用python二分法查找列表元素下标

小亿
122
2023-11-09 12:02:05
栏目: 编程语言

使用二分法查找列表元素的下标,可以按照以下步骤进行:

  1. 定义一个二分查找函数,接收一个有序列表和待查找的元素作为参数。函数的返回值为元素在列表中的下标,如果元素不存在,则返回-1。
  2. 在函数中初始化左边界为0,右边界为列表长度减1。
  3. 使用一个while循环,当左边界小于等于右边界时,进行以下操作:
    • 计算中间位置的索引值:mid = (left + right) // 2
    • 如果中间位置的元素等于待查找的元素,返回中间位置的索引值。
    • 如果中间位置的元素大于待查找的元素,将右边界更新为mid - 1。
    • 如果中间位置的元素小于待查找的元素,将左边界更新为mid + 1。
  4. 如果循环结束后仍然没有找到元素,返回-1。

下面是一个使用二分法查找列表元素下标的示例代码:

def binary_search(arr, target):
    left = 0
    right = len(arr) - 1

    while left <= right:
        mid = (left + right) // 2
        if arr[mid] == target:
            return mid
        elif arr[mid] < target:
            left = mid + 1
        else:
            right = mid - 1

    return -1

# 示例输入和输出
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
target = 6
index = binary_search(arr, target)
print(index)  # 输出: 5

在以上示例中,通过调用binary_search函数,查找列表arr中元素target的下标,并将结果打印出来。

0
看了该问题的人还看了