剑指offer-python 数组中数值和下标相等的元素

题目描述:

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

样例:

输入:[-3, -1, 1, 3, 5]
输出:3
注意:如果不存在,则返回-1。

分析:

首先想到的方法是对数组从头遍历,当遍历到一个元素和其下标相等时直接返回,如果遍历结束仍没有找到则返回-1。这种方法的时间复杂度是O(n)。
第二种方法考虑输入数据的特点:单调递增且唯一,所以使用二分查找。

  • 如果mid=nums[mid],则返回mid;

  • 如果mid>nums[mid],则mid左侧的元素肯定不再匹配。

    例如[-1,0,1,3,5]

  • 如果mid<nums[mid],则mid右侧的元素肯定不再匹配。

    例如[0,1,3,4,5]

class Solution(object):
    def getNumberSameAsIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        
        sta=0
        end = len(nums)-1
        while sta<=end:
            mid = sta+(end-sta)//2
            if mid == nums[mid]:
                return mid
            elif mid>nums[mid]:
                sta = mid+1
            else:
                end = mid-1
        return -1
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值