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

解法:二分

根据数组元素和下标的关系,进行二分即可
时间复杂度O(logn),空间复杂度O(1)

  1. class Solution {
  2. public:
  3. int getNumberSameAsIndex(vector<int>& nums) {
  4. if (nums.empty()) return -1;
  5. int l = 0, r = nums.size() - 1;
  6. while (l < r) {
  7. int mid = l + r >> 1;
  8. if (nums[mid] >= mid) r = mid;
  9. else l = mid + 1;
  10. }
  11. if (nums[l] != l) return -1;
  12. return l;
  13. }
  14. };