题目:在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数(原题目以及示例为下方链接)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:从右上角开始查找,两层for循环查找,外层for循环为记录行,内层for循环记录列:
若目标值等于二维数组右上角的值,则直接找到,退出两层的for循环;
若目标值大于二维数组右上角的值,则跳出内层for循环,记录当时的列号,并使行号加1,继续进入下一次的外层for循环及内层for循环,即从该位置(行号加1、记录的列号)处继续循环。
若目标值小于二维数组右上角的值,则跳出内层for循环,记录当时的行号,并使列号减1,进入下一次的内层for循环。
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int> >& matrix, int target) {
//必须首先判断是否为空的数组
if(matrix.empty()){
return false;
exit(0);
}
int n = matrix.size(); //行数
//cout << "n= " << n << endl;
int m = matrix[0].size(); //列数
//cout << "m= " << m << endl;
int flag = 0; //标记,使跳出两层的for循环
int i = 0;
int j = m - 1;
for (; i < n; i++) {
for (; j > -1; j--) {
if (target == matrix[i][j]) {
flag++;
//cout << "找到了这个数 " << matrix[i][j] << endl;
break;
}
else if (target > matrix[i][j]) {
break;
}
else {
continue;
}
}
if (flag != 0) {
break;
}
}
if (flag != 0) {
return true;
}
else {
return false;
}
}
};
注意:提交时,会出现错误,Line 1033: Char 9: runtime error: reference binding to null pointer of type 'std::vector<int, std::allocator<int>>' (stl_vector.h)
这是因为测试代码可能为一个空的二维数组,所以需要首先判断是否为空的数组。