题目描述
- 在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
算法分析
- 矩阵是有序的,从右上角来看,向下数字递增,向左数字递减。因此从右上角开始查找,当要查找数字比右上角数字大时,下移;要查找数字比右上角数字小时,左移。
- 注意边界条件判断。
提交代码:
class Solution {
public:
bool Find(int target, vector<vector<int>> array) {
if (array.empty() || array[0].empty())
return false;
int row = 0;
int col = array[0].size() - 1;
while (row < array.size() && col >= 0)
{
if (array[row][col] == target)
return true;
else if (target > array[row][col])
++row;
else if (target < array[row][col])
--col;
}
return false;
}
};
测试代码:
#include<iostream>
#include<vector>
using namespace std;
/*
// 面试题4:二维数组中的查找
// 题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按
// 照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个
// 整数,判断数组中是否含有该整数。
*/
// ====================测试代码====================
void Test(char* testName, vector<vector<int>> matrix, int number, bool expected)
{
if (testName != nullptr)
printf("%s begins: ", testName);
Solution s;
bool result = s.Find(number, matrix);
if (result == expected)
printf("Passed.\n");
else
printf("Failed.\n");
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数在数组中
void Test1()
{
vector<vector<int>> matrix = { { 1, 2, 8, 9 },{ 2, 4, 9, 12 },{ 4, 7, 10, 13 },{ 6, 8, 11, 15 } };
Test("Test1", matrix, 7, true);
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数不在数组中
void Test2()
{
vector<vector<int>> matrix = { { 1, 2, 8, 9 },{ 2, 4, 9, 12 },{ 4, 7, 10, 13 },{ 6, 8, 11, 15 } };
Test("Test2", matrix, 5, false);
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数是数组中最小的数字
void Test3()
{
vector<vector<int>> matrix = { { 1, 2, 8, 9 },{ 2, 4, 9, 12 },{ 4, 7, 10, 13 },{ 6, 8, 11, 15 } };
Test("Test3", matrix, 1, true);
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数是数组中最大的数字
void Test4()
{
vector<vector<int>> matrix = { { 1, 2, 8, 9 },{ 2, 4, 9, 12 },{ 4, 7, 10, 13 },{ 6, 8, 11, 15 } };
Test("Test4", matrix, 15, true);
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数比数组中最小的数字还小
void Test5()
{
vector<vector<int>> matrix = { { 1, 2, 8, 9 },{ 2, 4, 9, 12 },{ 4, 7, 10, 13 },{ 6, 8, 11, 15 } };
Test("Test5", matrix, 0, false);
}
// 1 2 8 9
// 2 4 9 12
// 4 7 10 13
// 6 8 11 15
// 要查找的数比数组中最大的数字还大
void Test6()
{
vector<vector<int>> matrix = { { 1, 2, 8, 9 },{ 2, 4, 9, 12 },{ 4, 7, 10, 13 },{ 6, 8, 11, 15 } };
Test("Test6", matrix, 16, false);
}
// 鲁棒性测试,输入空指针
void Test7()
{
Test("Test7", vector<vector<int>>(), 16, false);
}
int main(int argc, char* argv[])
{
Test1();
Test2();
Test3();
Test4();
Test5();
Test6();
Test7();
return 0;
}