题意
假设一个单调递增的数组里的每个元素都是整数并且是唯一的。
请编程实现一个函数找出数组中任意一个数值等于其下标的元素。
例如,在数组 [−3,−1,1,3,5]
中,数字 3 和它的下标相等。
分析
简单二分。区间性质 nums[i] >= i, 找到满足这个条件的第一个下标 index。
如果 nums[index] == index, 那么index 就是答案,否则不存在这样的数。
class Solution {
public:
int getNumberSameAsIndex(vector<int>& nums) {
int index = find(nums);
if(nums[index] == index)return index;
else return -1;
}
int find(vector<int>& nums){
int l = 0, r = nums.size()-1;
while(l < r){
int mid = l+r>>1;
if(nums[mid] >= mid)r = mid;
else l = mid + 1;
}
return l;
}
};