前言
近期在刷LeetCode上一道题时,遇到了一点点问题,做个小笔记。
剑指 Offer 04. 二维数组中的查找:
在一个 n * m的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个高效的函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
一、解题思路
从左下角开始走,利用这个顺序关系可以在O(m+n)的复杂度下解决这个题:
如果当前位置元素比target大,则row–
如果当前位置元素比target小,则col++
如果相等,返回true
如果越界了还没找到,说明不存在,返回false
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
int i = matrix.size() - 1, j = 0;
while (i >= 0 && j < matrix[0].size())
{
if (matrix[i][j] > target)
i--;
else if (matrix[i][j] < target)
j++;
else return true;
}
return false;
}
};
int main() {
Solution s;
vector<vector<int>> matrix{ };
int target = 5;
bool ans = s.findNumberIn2DArray(matrix, target);
cout << boolalpha << ans << endl;
system("pause");
}
二、错误
1.错误示范
一开始我是从右下角开始遍历的,思路同样也没有问题,但是败在细节。
从右上角开始走,利用这个顺序关系可以在O(m+n)的复杂度下解决这个题:
如果当前位置元素比target小,则row++
如果当前位置元素比target大,则col–
如果相等,返回true
如果越界了还没找到,说明不存在,返回false
错误代码如下(示例):
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
int r = 0;
int c = matrix[0].size() - 1 ;
while(r < matrix.size() && c >= 0){
if(matrix[r][c] > target)
c--;
else if(matrix[r][c] < target)
r++;
else
return true;
}
return false;
}
};
乍一看代码跟第一部分从左下角开始遍历没有区别,但是编译总是出现错误:
Line 1033: Char 9: runtime error: reference binding to null pointer of type ‘std::vector<int, std::allocator>’ (stl_vector.h)
SUMMARY: UndefinedBehaviorSanitizer: undefined-behavior /usr/bin/…/lib/gcc/x86_64-linux-gnu/9/…/…/…/…/include/c++/9/bits/stl_vector.h:1043:9
2.错误分析
仔细分析,原来是因为vector容器size()的使用出现了问题:
LeetCode的测试中还包括二维数组为空[ ] ,在此时matrix.size()会显示为0,而matrix[0].size()就会报错!!!
代码如下(示例):
#include<iostream>
#include<vector>
using namespace std;
int main() {
vector<vector<int>>arr{};
int a = arr.size();//注意
cout << a << endl;
system("pause");
}
运行正常,结果为0.
该处使用的为arr[0].size()。
#include<iostream>
#include<vector>
using namespace std;
int main() {
vector<vector<int>>arr{};
int a = arr[0].size(); //注意
cout << a << endl;
system("pause");
}
会报错vector容器超出范围!!!
该处使用的为arr[0].size(),但是vector为{{ }}!!!
int main() {
vector<vector<int>>arr{ { } };// !!!
int a = arr[0].size();
cout << a << endl;
system("pause");
}
结果为0,不会报错。
而在第一部分的代码中:
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
int i = matrix.size() - 1, j = 0;
while (i >= 0 && j < matrix[0].size())
{
if (matrix[i][j] > target)
i--;
else if (matrix[i][j] < target)
j++;
else return true;
}
return false;
}
};
实际上并没有进行到while循环中判断 j < matrix[0].size()这一步,因为如数组为空,i = -1,不满足i >= 0的条件,实际上在此种条件下没有进入while循环。
3.错误改正
从右下角开始遍历时,加上判断vector二维数组是否为空的条件。
代码如下(示例):
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
if(matrix.size() < 1 || matrix[0].size() < 1 )
return false;
int r = 0;
int c = matrix[0].size() - 1 ;
while(r < matrix.size() && c >= 0){
if(matrix[r][c] > target)
c--;
else if(matrix[r][c] < target)
r++;
else
return true;
}
return false;
}
};
总结
1.注意二维数组中vector为{ }以及{ { } }的情况。其中
- vector为{ },matrix[0].size()报错。
- vector为{ { } },matrix[0].size()为0。
一般情况下,这两种情况LeetCode都会测试。
2.即使vector为{ },但是matrix[0].size()用在if( )条件判断中还是可以用的。