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

2020年2月16日 评论 287 views 1251字阅读4分10秒

前言

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

题目位置:力扣中国

题目

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

思路

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

举例

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

剑指Offer(一):二维数组中的查找

查找过程如下:

剑指Offer(一):二维数组中的查找

代码

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

引用

https://cuijiahua.com/blog/2017/11/basis_1.html

小熊