题目:
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解法:
从右上角
或者左下角
的两个点开始,每一步都能排除一行或者一列
代码:
#include <iostream>
using namespace std;
bool Find(int* matrix, int rows, int columns, int number) {
if (matrix != nullptr && rows > 0 && columns > 0) {
int row = 0;
int col = columns - 1; //右上角或者左下角
while (row < rows && columns >= 0)
{
if (matrix[row * columns + col] > number)
{
col--;
}
else if (matrix[row * columns + col] < number)
{
row++;
}
else
{
return true;
}
}
}
return false;
}
int main()
{
int matrix[4][4] = { {1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15} };
cout << Find((int*)matrix, 4, 4, 7);
}
备注:
- C++二维数组定义或者传参都需要指定每行的个数(第二维度)
- 多维数组传参需要使用
(int*)
转换为int指针,并且元素访问按照一维数组的方式(matrix[row * columns + col]
) - 时间复杂度
O(row + column)