思路:有序+查找元素,第一想到的就是二分法,所以我对每行采用二分法,每行的复杂度为O(m),那么总的复杂度就是O(nlogm);看题解从右上角开始,这个数比目标值大,就往左移一列,比目标值小,就往下移一列,正式利用了整个二维数组从左到右从上到下的递增性;
第二种方法还是比较容易实现的,这里就先写第一种方法,回顾以下二分法;
bool binary_search(vector<int> &a, int target)
{
int len = a.size();
int l = 0, r = len - 1;
while (l <= r)
{
int mid = (l + r) / 2;
if (l == r && a[l] != target)
{
return false;
}
if (a[mid] == target)
{
return true;
}
else if (a[mid] < target)
{
l = mid + 1;
}
else
{
r = mid - 1;
}
}
}
bool findNumberIn2DArray(vector<vector<int>> &matrix, int target)
{
//先用二分法找到所在行,再用二分法找到所在列
int len_row = matrix.size();
for (int i = 0; i < len_row; i++)
{
if (binary_search(matrix[i], target))
return true;
}
return false;
}