剑指Offer面试题04. 二维数组中的查找

小熊

310
文章

12
评论

2020年2月16日剑指Offer1 3,567字数 1251阅读4分10秒阅读模式

前言

本系列文章为《剑指Offer》刷题笔记。

题目位置：力扣中国

题目

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

思路

首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数组，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

举例

如果在一个二维数组中找到数字7，则返回true，如果没有找到，则返回false。

查找过程如下：

代码

C++：

``` 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;
}
};

<pre><code class="line-numbers">Python2.7：

``` python
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
rows = len(array)
cols = len(array[0])
if rows > 0 and cols > 0:
row = 0
col = cols - 1
while row < rows and col >= 0:
if target == array[row][col]:
return True
elif target < array[row][col]:
col -= 1
else:
row += 1
return False

Python2.7 解法2

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self,target, array):
        # write code here
        xflag = 0
        for row in array:
            if xflag == 0 :
                xflag = len(row)-1
            while xflag >= 0:
                if target < row[xflag]:
                    xflag = xflag - 1
                elif target > row[xflag]:
                    break
                else:
                    return True
        return False