题目
在一个 n * m 的二维数组中,每一行都按照从左到右 非递减 的顺序排序,每一列都按照从上到下 非递减 的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路
在数组中找数字,显而易见我们可以采用哈希法,但这里是个特殊数组,左上角一定是最小数字,可以用这个特性采用模拟法解题,大致思路如下
自右上角开始,对比目标值,如若比目标值大则向左移动,如若比目标值小则向下移动,直到遍历到边界条件或者找到符合为止
代码如下:
//模拟法 从右上角出发
#include <vector>
using namespace std;
class Solution
{
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target)
{
bool result = false;
if (matrix.empty())return result;
int max_x = matrix.size();
int max_y = matrix[0].size();
int x = 0;
int y = max_y - 1;
while (x < max_x && y >= 0)
{
if (matrix[x][y] == target)
{
result = true;
break;
}
else if (matrix[x][y] > target)
{
--y;
}
else
{
++x;
}
}
return result;
}
};
这里尤其要注意处理matrix为空的情况,用了empty处理抛出