题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
分析:
自然容易想到二分法,但二分法不是最优解,利用元素特殊的排列进行删选(每次选取右上角或者左下角的数字)
以下列二维数组为例 ,从9开始,如果9>target,那么第四列整列都比target大,可以删掉,col-1;如果9<target,那么第一行整行都比target小,可以删掉,rol+1。以此类推直到找到等于target的数。
代码:
c++版本:
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int rows = array.size();
int cols = array[0].size();
if(!array.empty() && rows > 0 && cols > 0){
int row = 0;
int col = cols - 1;
while(row < rows && col >= 0){
if(array[row][col] == target){
return true;
}
else if(array[row][col] > target){
--col;
}
else{
++row;
}
}
}
return false;
}
};
Python版本:
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
row =0
col = len(array[0])-1
while col>=0 and row<len(array):
if target > array[row][col]:
row += 1
elif target < array[row][col]:
col -= 1
else:
return True