[剑指offer] 数组中数值和下标相等的元素

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

题目

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

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

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

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

样例

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

输出：3

题解

思路

• 二分法

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