python

超轻量级php框架startmvc

Python查找数组中数值和下标相等的元素示例【二分查找】

更新时间:2020-06-24 07:06:01 作者:startmvc
本文实例讲述了Python查找数组中数值和下标相等的元素。分享给大家供大家参考,具体如下

本文实例讲述了Python查找数组中数值和下标相等的元素。分享给大家供大家参考,具体如下:

题目描述:

假设一个单调递增的数组中的每个元素都是整数并且是唯一的。请编程实现一个函数,找出数组中任意一个数值等于其下标的元素,例如在数组【-3,-1,1,3,5】中,3和他的下标相等。

采用二分查找:如果数组中的数字小于下标,由于下标是-1的递减数列,但是数组中的元素差值大于等于-1,因此左边的不可能等于下标。如果数组中的数字大于下标,同理,之后的数字肯定都大于下标,往左边查找。

算法示例:


# -*- coding:utf-8 -*-
#! python3
class Solution:
 def numberEqualSubscript(self, numbers):
 if numbers == []:
 return -1
 left = 0
 right = len(numbers) - 1
 while(left <= right):
 middle = (left + right) >> 1
 if numbers[middle] == middle:
 return middle
 elif numbers[middle] < middle:
 left = middle + 1
 else:
 right = middle - 1
 return -1
numbers = [-3,-1,1,3,5]
print(Solution().numberEqualSubscript(numbers))

运行结果:

3

Python 查找 数组 数值 下标 相等 元素 二分查找