《编程之美》上的一个问题:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
public:
bool Find(vector<vector<int> > array,int target) {
//.......
}
要解决这个问题,我相信很多人的第一想法是可以用递归,首先将target和array(m*n)左下角的元素array[m][0]比较:如果target<array[m][0],就可以排除最后一行,在剩下的array1(m-1*n)中寻找;如果target>array[m][0],就可以排除第一列,在剩下的array1(m*n-1)中寻找。同理可以递归下去......
想法很可行,但是去掉一行或去掉一列这样的操作对vector<vector<int> > array的操作太复杂,实现起来不太容易。
再思考得到这样的算法:
bool Find(vector<vector<int> > array,int target) {
int x = array.size();
if(x == 0)
return false;
x--;
int y = 0;
while(x >= 0 && y < array[0].size())
{
if( array[x][y] == target )
return true;
else if( array[x][y] < target )
y++;
else //array[x][y]>target
x--;
}
return false;
}
基本思想其实很类似于递归,但是是用x,y来标记位置,从左下角向右上角搜索,每次循环向上移动1,或者向右移动1,类似就排除一行或一列。最终会找到,或者到达终点右上角结束没找到。