第二道:有序二维数组中查找元素的问题
问题描述:在一个二维数组中每一行都按照从左到右的顺序递增,每一列也按照从上到下的顺序递增。输入一个二维数组和一个整数,判断数组中是否有该整数。
例:
1 | 2 | 8 | 9 |
2 | 4 | 9 | 12 |
4 | 7 | 10 | 13 |
6 | 8 | 11 | 15 |
查找7,返回true;查找5,返回false。
算法思路:1.双层循环,逐一比较,时间复杂度为O(n*n)
2.从中间选择一个元素num,若是比num大则向数组的左上方查找;若是比num小则向数组的右下方查找。思路简单,但是要查找的数字相对于当前位置有可能在两个 区域中出现,且这两个区域还有重叠,使得问题变得复杂。
3.选取数组的右上角或是左下角的数字和待查数字进行比较。以选取右上角,查找数字7进行举例。9 > 7,舍弃第四列;8 > 7,舍弃第三列;2 < 7,舍弃第一行;4 < 7,舍弃第二行;7 = 7,返回true。简单高效。
下面是第三种想法的实现(右上角为例)
bool Find(vector<vector<int> > array,int target) {
bool found = false;
int rows = array.size();
int columns = array[0].size();
if(!array.empty() && rows > 0 && columns > 0){
int row = 0;
int column = columns - 1;
while(row < rows && column >= 0){
if(array[row][column] == target){
found = true;
break;
}
else if(array[row][column] > target)
-- column;
else
++ row;
}
}
return found;
}