题目描述
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路
首先,根据这个二维数组array的特点,数组左上角元素array[0][0]是最小值,数组右下角元素array[m-1][n-1]是最大值(假设数组的大小是m*n)。
从右上角元素array[0][n-1]入手,将该元素与待查找的整数target进行对比。如果两者相同,则找到了target,返回true;如果该元素大于target,则该元素所在的列里的元素都将大于target,因此不考虑第n-1列,在剩下的m*(n-1)数组中寻找target;如果该元素小于target,则该元素所在的行里的元素都将大于target,因此不考虑第0行,在剩下的(m-1)*n数组中寻找target。
用两个指针i和j指示当前考虑的数组。一开始,令i=0,j=n-1,表示考虑从i=0到m-1,j=0到n-1的整个数组。每次寻找的过程,就是用array[i][j]与target进行比较。随着寻找过程的进行,i不断增大,j不断减小。这个过程一直进行到i>m-1或者j<0为止,如果此时还未找到target,则意味着二维数组array中不存在target。
代码
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int m = array.size(); //二维数组的长
if (m == 0)
return 0;
int n = array[0].size(); //二维数组的宽
if (n == 0)
return 0;
if (target < array[0][0] || target > array[m - 1][n - 1]) //超过最小和最大值
return 0;
int i = 0, j = n - 1; //从左上角开始寻找
while (i <= m - 1 && j >= 0)
{
if (target == array[i][j]) //与左上角元素相同,则找到
return 1;
else if (target > array[i][j]) //比左上角元素大,则删除第i行
i++;
else //比左上角元素小,则删除第j列
j--;
}
return 0; //没有找到
}
};