题目:
定义局部最小的概念。arr长度为1时,arr[0]是局部最小。arr的长度为N(N>1)时,如果arr[0]<arr[1],那么arr[0]是局部最小;如果arr[N-1]<arr[N-2],那么arr[N-1]是局部最小;如果0<i<N-1,既有arr[i]<arr[i-1]又有arr[i]<arr[i+1],那么arr[i]是局部最小。 给定无序数组arr,已知arr中任意两个相邻的数都不相等,写一个函数,只需返回arr中任意一个局部最小出现的位置即可。
测试用例:
输入:[3,2,1,4,5]
输出:2
思路:
根据题目:如果arr为空,那么直接返回false;如果arr只有一个元素,或者arr[0] < arr[1] ,那么直接返回arr[0];如果arr[-1] < arr[-2],那么直接返回arr[-1]。如果都没有找到,那么可以用二分查找的方法:首先,我们找到中间位置mid,假设arr[mid] < arr[mid-1] && arr[mid] < arr[mid+1],那么mid位置的元素就是一个局部最小值;否则若arr[mid] > arr[mid-1],那么从mid向左看去呈现递减的趋势,说明局部最小值可能在mid左边;若arr[mid] > arr[mid+1],那么从mid向右看去呈现递减的趋势,说明局部最小值可能在mid右边。
代码:
class Solution {
public:
int getLessIndex(vector<int> arr) {
if (arr.size() < 1) return -1;
if (arr.size() == 1 || arr[0] < arr[1]) return 0;
if (arr[arr.size() - 1] < arr[arr.size() - 2]) return arr.size() - 1;
int left = 0, right = arr.size() - 1;
int res = binarysearch(arr, left, right);
return res;
}
int binarysearch(vector<int> arr, int left, int right) {
if (left > right) return -1;
int mid = left + (right - left) / 2;
if (mid == 0 || mid == arr.size() - 1) return -1;
if (arr[mid] < arr[mid + 1] && arr[mid] < arr[mid - 1]) {
return mid;
}
if (arr[mid] > arr[mid - 1]) {
int res = binarysearch(arr, left, mid);
if (res != -1) {
return res;
}
}
if (arr[mid] > arr[mid + 1]) {
int res = binarysearch(arr, mid, right);
if (res != -1) {
return res;
}
}
return -1;
}
};
note:
如果确保了数组中一定存在峰值,那么做法就会简单很多。比如162. 寻找峰值 - 力扣(LeetCode)
该题是求数组中的局部最大值。在这题中,增加了一个条件:
- 对于所有有效的
i
都有nums[i] != nums[i + 1]
假设数组中不存在峰值 ,那么只有2种情况:
就是说,数组中所有数都相等,或者最大值的左(右)边也是最大值。但是因为有nums[i] != nums[i+1]的限定,以上情况不成了,所以数组中一定有峰值。那么这种情况下,代码可以是:
func FindPeakElement(nums []int) int {
left, right := 0, len(nums)-1
for left < right {
mid := left + (right-left)/2
if nums[mid] > nums[mid+1] {
right = mid
} else {
left = mid + 1
}
}
return left
}