编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target。该矩阵具有以下特性:
每行的元素从左到右升序排列。
每列的元素从上到下升序排列。
示例:
现有矩阵 matrix 如下:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。
采用的二分法。整个矩阵分成了M*N一维数组,拆分为在for循环下的一维数组。题意就成为了一维数组中查找。二分法将数组一分为二。判断在左边还是右边。接着缩短边界在二分。直到找到。
public class Solution {
public bool SearchMatrix(int[,] matrix, int target) {
if (matrix.Length == 0) return false;
int row = matrix.GetLength(0);
int col = matrix.GetLength(1);
for (int i = 0; i < row; i++)
{
int left = 0;
int right = col - 1;
while (left <= right)
{
int mid = (left + right) / 2;
if (matrix[i, mid] == target) return true;
if (matrix[i, mid] < target) left = mid + 1;//说明target不在左边
else right = mid - 1;//说明target不在右边边
}
}
return false;
}
}
第二种方法。数据是有序排列,采用剪枝,选取起点
选左上角,往右走和往下走都增大,不能选
选右下角,往上走和往左走都减小,不能选
选左下角,往右走增大,往上走减小,可选
选右上角,往下走增大,往左走减小,可选
public class Solution {
public bool SearchMatrix(int[,] matrix, int target) {
if (matrix.Length == 0) return false;
int row = matrix.GetLength(0)-1;
int col = 0;
while (row>=0 &&col<matrix.GetLength(1))
{
if (matrix[row,col] == target)
{
return true;
}
else if (matrix[row,col] > target)
{
row--;
}
else if(matrix[row,col] < target)
{
col++;
}
}
return false;
}
}