LeetCode之剑指offer-二维数组的查找

学习记录，仅供参考。

题目描述：

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

   [                                       target = 5 返回Trus    target = 20 返回False
     [1,   4,  7, 11, 15],
     [2,   5,  8, 12, 19],
     [3,   6,  9, 16, 22],
     [10, 13, 14, 17, 24],
     [18, 21, 23, 26, 30]
   ]

拿到题目首先想到的肯定是暴力查找，我遍历一次二位数组，肯定可以。

def findNumberIn2DArray(self, matrix, target):
        if len(matrix) == 0:
            return False
        n, m = len(matrix), len(matrix[0])
        count = 0
        for i in range(n):
            for j in range (m):
                pos = matrix[i][j]
                if target == pos:
                    count += 1
        if count >= 1:
            return True
        else:
            return False

提交之后确实通过了，再来学习一下liweiwei大佬的算法。

    def findNumberIn2DArray(self, matrix, target):
        rows = len(matrix)
        if rows == 0:
            return False

        cols = len(matrix[0])
        if cols == 0:
            return False

        # 从右上角开始查找
        x = 0
        y = cols - 1
        while x < rows and y >= 0:
            if target == matrix[x][y]:
                return True
            elif target < matrix[x][y]:
                y -= 1
            else:
                x += 1
        return False

解释：（减而治之）

暴力查找固然可以解决，但是没有用到题目的特点，题目强调了矩阵从上到下递增，从左到右递增。

如果从左上角开始找，根据题目特点，遇到的数字都是逐渐增大的；

如果从右下角开始找，根据题目特点，遇到的数字都是逐渐减小的；

2	4	5	9	14
3	5	7	12	15
4	6	8	13	16
5	7	9	14	17
6	8	10	15	18

如果target = 10

从右上角14开始查找，判断10不等于14且10小于14那么行x不变列y-1，移动到9；

10不等于9且大于9，行x+1且列不变，移动到12；

10不等于12且小于12，行x不变列y-1，移动到7；

之后同理，会一直按照这样的规则，沿着红色路径寻找，直到找到10，或者没有相同数字，搜索结束返回False。

选择右上角或者左下角，搜索路径都是不回头的。

或者可以理解为，当判断target小于当前数，那么target一定在当前数左边，消去当前数的这一列，如果target大于当前数，那么target一定在当前数的下面，消去当前数的这一行，所有这种算法肯定是能查全的。详细图解可见leetcode（https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/solution/mian-shi-ti-04-er-wei-shu-zu-zhong-de-cha-zhao-zuo/）