题目地址:
https://www.acwing.com/problem/content/description/65/
假设一个单调递增的长 n n n数组里的每个元素都是整数并且是唯一的。请编程实现一个函数找出数组中任意一个数值等于其下标的元素。例如,在数组 [ − 3 , − 1 , 1 , 3 , 5 ] [−3,−1,1,3,5] [−3,−1,1,3,5]中,数字 3 3 3和它的下标相等。
思路是二分。可以看出,如果一个数是小于其下标的,那么其左边的数都将小于下标;如果一个数是大于等于其下标的,那么其右边的数都将大于等于下标。所以只需找到第一个大于等于其下标的数,看一下这个数符不符合条件即可。可以用二分。代码如下:
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int getNumberSameAsIndex(vector<int>& nums) {
int l = 0, r = nums.size() - 1;
while (l < r) {
int m = l + (r - l >> 1);
if (nums[m] >= m) r = m;
else l = m + 1;
}
return nums[l] == l ? l : -1;
}
};
时间复杂度 O ( log n ) O(\log n) O(logn),空间 O ( 1 ) O(1) O(1)。