题目
思路
解法一
首先用最简单的方法,也就是暴力枚举。这里使用了LINQ,只需要一行代码,但等价于写两个for循环。
public bool FindNumberIn2DArray(int[][] matrix, int target)
{
return matrix.Any(t => t.Any(t1 => t1 == target));
}
解法二
解法一完全没有考虑二维数组元素已经排好序的条件,这次我们将其考虑进来。对于已经排好序的数组,我们知道二分法是最高效的方法,对于这道题目是否能使用二分法呢?
首先我们找到二维数组最中间的数字,也就是9。那么假设我们寻找的是5,此时5<9,因此对于9右下方的数组都不可能存在5,从而排除掉了一片区域。
现在我们知道,5有可能存在于9的左侧或上方这两片区域,且这两片区域有所重叠。下一步又如何二分呢?似乎有些困难。那我们不妨退一步,一列一列地进行排除。
由于二维数组每一列的顺序都是从小到大排序,所以我们可以从数组的右上角开始对比。假如这个数字比target大,那么这一整列都可以排除掉。
当我们遍历到4,此时已经比5小,也就是说这一整行都可以排除,因为这一行前面的数字都比4小。
继续对比右上角的数字,发现正好等于5,说明该数组存在目标元素,返回true。
代码如下
public bool FindNumberIn2DArray2(int[][] matrix, int target)
{
if (matrix.Length == 0)
return false;
if (matrix[0].Length == 0)
return false;
int m = matrix.Length;
int row = 0, col = matrix[0].Length - 1;
while (row < m && col >= 0)
{
// 右上角元素大于目标值,排除该列
if (matrix[row][col] > target)
{
col--;
}
// 右上角元素小于目标值,排除该行
else if(matrix[row][col] < target)
{
row++;
}
else
{
return true;
}
}
return false;
}