渐近思想——寻找局部最小值
数组的局部最小元素
给定一个长度为N的数组
a
,找到一个局部最小元素:满足a[i-1]>a[i]<a[i+1]
的索引i
;要求时间复杂度为O(lgN)。前置条件:数组中不含有重复的元素。
思路
首先,要注意到如果没有前置条件,那么就不可能在O(lgN)的限制内解决该问题。因为最起码需要将数组遍历一遍来确定是不是真的有可能存在局部最小值(比如全是1的情况下就不存在)。
接着,我们将数组拓展为函数,那么对于某个点A来说,存在另一个点B比它的值小,那么沿着AB方向一定存在一个局部最小值。
最后,基于这个思路,我们用二分查找来进行寻找。
- 这里的点A就是二分点、四分点、八分点…;
- 通过比较点A
a[mid]
与a[mid-1]
和a[mid+1]
的大小关系来决定寻找的方向,即新的点A的位置; - 局部最小值一定存在于某个分区间内,每次二分时,点A都在逐渐逼近局部最小值;
代码
public static int find(int[] a, int start, int end) {
int mid = (start + end) / 2;
if(mid < 0 || mid > a.length - 1) {return -1;}
if(a[mid-1] > a[mid] && a[mid] < a[mid+1]) {return a[mid];}
else if(a[mid-1] < a[mid]) {return find(a, start, mid);}
else {return find(a, mid, end);}
}
矩阵的局部最小元素
对于一个N*N的二维数组
a
,找到矩阵中的局部最小元素:满足a[i][j] < a[i][j+1]
,a[i][j] < a[i+1][j]
,a[i][j] < a[i][j-1]
,a[i][j] < a[i-1][j]
;
要求复杂度O(N);
思路
同样,利用渐近思想。想象你处在一片崇山峻岭之中,试图找到一个低谷。那自然是按顺序(如顺时针)考察是否有下坡路,到达下一个位置后重复此过程。
另外,还需要注意边界。
代码
public static int getLocalMinEle(int[][] a) {
// 获取二维数组的行列数
int rowSize = a.length;
int colSize = a[0].length;
int i, j; // 当前考察的元素坐标
int ans = a[0][0];
for (i = 0, j = 0; ; ) {
System.out.println("" + i + " " + j);
if ((j < rowSize-1) && a[i][j] > a[i][j+1]) { //右
j++;
continue;
}
if ((i< rowSize - 1) && a[i][j] > a[i + 1][j]) { //下
i++;
continue;
}
if ((j > 0) && a[i][j] > a[i][j - 1]) { //左
j--;
continue;
}
if ((i > 0) && a[i][j] > a[i - 1][j]) { //上
i--;
continue;
}
ans = a[i][j];
break;
}
return ans;
}