题目
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路
首先选取数组中右上角的数字。如果该数字等于要查找的数字,查找过程结束;如果该数字大于要查找的数组,剔除这个数字所在的列;如果该数字小于要查找的数字,剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角,则每一次都在数组的查找范围中剔除一行或者一列,这样每一步都可以缩小查找的范围,直到找到要查找的数字,或者查找范围为空。
#include <iostream>
#include <vector>
using namespace std;
class Solution
{
public:
bool findIn2vector(vector < vector <int> > num, int target)
{
if (num.size() == 0)
return false;
int rows = num.size();
int cols = num[0].size();
int row = 0;
int col = cols - 1;
while (row < rows && col >=0)
{
if (num[row][col] == target)
return true;
else if ( target < num[row][col])
col--;
else
row++;
}
return false;
}
};
int main()
{
Solution s;
vector <vector <int> > number{
{1, 2, 8, 9},
{2, 4, 9, 12},
{4, 7, 10, 13},
{6, 8, 11, 15} };
cout << s.findIn2vector(number, 7) << endl;
system("pause");
return 0;
}