AcWing 69. 数组中数值和下标相等的元素

题目描述

假设一个单调递增的数组里的每个元素都是整数并且是唯一的。

请编程实现一个函数找出数组中任意一个数值等于其下标的元素。

例如，在数组[-3, -1, 1, 3, 5]中，数字3和它的下标相等。

样例

输入：[-3, -1, 1, 3, 5]

输出：3

注意:如果不存在，则返回-1。

问题分析 

由于数组是单调递增的，所以可以用二分查找。通过比较数值和下标的大小关系来二分查找，如果当前位置的数值比下标小说明目标在不包括当前位置的右侧，否则说明在包括当前位置的左侧。返回时要判断一下，如果找到的位置和该位置上的数值相等则返回，如果不相等说明不存在这样的元素，那么返回-1。

代码实现

class Solution {
public:
    int getNumberSameAsIndex(vector<int>& nums) {
        int l = 0, r = nums.size() - 1;
        while(l < r){
            int mid = l + r >> 1;
            if(nums[mid] - mid < 0)
                l = mid + 1;
            else
                r = mid;
        }
        return nums[l] == l ? l : -1;
    }
};