【剑指offer题解】二维数组中的查找

Rude3Knife的公众号

发布于 2019-12-02 13:41:09

51000

代码可运行

文章被收录于专栏：后端技术漫谈后端技术漫谈

运行总次数：0

代码可运行

题目介绍

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

解题思路

方法一

首先能够想到的肯定是一行一行或者一列一列遍历，判断数组中是否含有该整数。该方法显然是最笨拙的二维数组遍历，面试官也不会满意，时间复杂度是O(n^2)

代码

Python

class Solution:
    def Find(self, target, array):
        for i in range(len(array)):
            for j in range(len(array[0])):
                if array[i][j]==target:
                    return True
        return False

方法二

我们思考下，既然大的数字都在靠右边/下边，能否能利用行列的数据变化规律来优化下解法，如果寻找的目标数大于现在的数字，那么目标数字是在当前位置的右边或下边，如果所寻找的目标数小于现在的数字，那么目标数字在当前位置的左边或上边。举个例子，如下图数组所示：

1  2  3   4
2  3  8   9
3  4  9   10
4  5  10  11

我们的位置是1，要找8，8大于1，那么在1的右边和下边区域进行下一步的搜索。

如果我们先搜索他的右边，

又搜索了他们的下边，

2  3  8   9
3  4  9   10
4  5  10  11

这时候我们发现

3  8  9
4  9  10
5  10 11

这个区域搜索了两次，我们是从数组的第一个数[0][0]取的，遇到了重复搜索区域的问题。有没有方法去除重复的搜索区域呢，我们发现，当从右上角取第一个数的时候，可以去除重复的搜索区域，还是以这个数组为例，取4，搜索8，发现8比4大，那么8不可能出现在4这一行，只需要从下边搜索即可。如果寻找3，3比4小，4那一列后续的数都大于4，只需要从左边搜索即可。

1  2  3   4
2  3  8   9
3  4  9   10
4  5  10  11

我们还可以发现左下角的点也可以去除重复搜索区域，总结起来的话，有点像变量控制法的感觉，将一个变量控制住，根据另外一个变量的规律，得出结论。

这样我们将时间复杂度降为了O(n)

代码

Java

public class Solution {
    public boolean Find(int target, int [][] array) {
        int rows = array.length;
        int cols = array[0].length;
        int i = rows-1, j = 0;
        while(i >= 0 && j < cols){
            if(target < array[i][j])
                i--;
            else if(target > array[i][j])
                j++;
            else
                return true;
        }
        return false;
    }
}

Python

class Solution:
    def Find(self, target, array):
        if array == ([] or [[]]):
            return False
        row = len(array)
        col = len(array[0])
        i=0
        j=col-1
        while 0<=i<row and 0<=j<col:
            if target>array[i][j]:
                i=i+1
            elif target<array[i][j]:
                j=j-1
            else:
                return True
        return False

总结

题目简单，小伙伴们需要理解的是算法题应该注重的是效率优化，暴力穷举能够解决问题，但说服不了面试官。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2019-12-02，如有侵权请联系 cloudcommunity@tencent.com 删除

python

编程算法

本文分享自后端技术漫谈微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

python

编程算法

登录后参与评论

0 条评论

热度