题目描述
假设一个单调递增的数组里的每个元素都是整数并且是唯一的。
请编程实现一个函数找出数组中任意一个数值等于其下标的元素。
例如,在数组[-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;
}
};