其他分享
首页 > 其他分享> > 数组-剑指offer3数组中重复的数字-中等-20210807

数组-剑指offer3数组中重复的数字-中等-20210807

作者:互联网

数组-剑指offer3数组中重复的数字-中等-20210807

1. 题目描述

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

示例

[
  [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]
]

给定 target = 5,返回 true。
给定 target = 20,返回 false。

2. 题目解答

2.1 第一次尝试答案-暴力解法

时间复杂度:O(nm)

空间复杂度:O(1)

思路:

class Solution(object):
    def findNumberIn2DArray(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: bool
        """
        # 应考虑特殊情况
        if not matrix or len(matrix)==0 or len(matrix[0])==0 :
            return False
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if matrix[i][j] == target:
                    return True
        return False # bug:返回false应该在判断完所有之后的循环之外

2.2 参考解法(较优)

时间复杂度:O(m+n)

空间复杂度:O(1)

思路:利用已排序的特征

从二维数组的右上角开始查找。如果当前元素等于目标值,则返回 true

如果当前元素大于目标值,则移到左边一列(当前元素的下边的所有元素都一定大于目标值);如果当前元素小于目标值,则移到下边一行(说明当前元素的左边的所有元素都一定小于目标值)。

class Solution(object):
    def findNumberIn2DArray(self, matrix, target):
        '''
        从右上角
        :param matrix: int
        :param target: int
        :return: bool
        '''
        #  illegal input
        if not matrix or len(matrix) == 0 or len(matrix[0]) == 0:
            return False
        #  find target num
        row, column = 0, len(matrix[0])-1
        while row < len(matrix) and column >= 0:
            if matrix[row][column] == target:
                return True
            elif matrix[row][column] > target:
                column -= 1
            else:
                row += 1
        return False

    def findLeft(self, matrix, target):
        '''
        从左下角开始找
        :param matrix: 
        :param target: 
        :return: 
        '''
        if not matrix or len(matrix) == 0 or len(matrix[0]) == 0:
            return False
        row, column = len(matrix)-1, 0
        while column < len(matrix[0]) and row >= 0:
            if matrix[row][column] == target:
                return True
            elif matrix[row][column] > target:
                row -= 1
            else:
                column +=  1
        return False


if __name__ == '__main__':
    matrix1 = [[]]
    matrix2 = [
        [1, 2, 8, 9],
        [2, 4, 9, 12],
        [4, 7, 10, 13],
        [6, 8, 11,15]
    ]

    # Test case
    Solution = Solution()
    print(Solution.findNumberIn2DArray(matrix1, 7))
    print(Solution.findNumberIn2DArray(matrix2, 7))
    print(Solution.findNumberIn2DArray(matrix2, 15))
    print(Solution.findNumberIn2DArray(matrix2, 1))
    print(Solution.findNumberIn2DArray(matrix2, 20))
    print(Solution.findNumberIn2DArray(matrix2, 0))
    print(Solution.findNumberIn2DArray(matrix2, 3))

    print(Solution.findLeft(matrix1, 7))
    print(Solution.findLeft(matrix2, 7))
    print(Solution.findLeft(matrix2, 15))
    print(Solution.findLeft(matrix2, 1))
    print(Solution.findLeft(matrix2, 20))
    print(Solution.findLeft(matrix2, 0))
    print(Solution.findLeft(matrix2, 3))

3. 测试用例

1. 空数组
2. 二维数组中包含查找的数字(查找的数字是数组中的最大值/最小值/中间值)
3. 二维数组不包含查找的数字(大于/小于/处于中间但没有该值)

标签:target,Solution,数组,offer3,20210807,print,matrix2,matrix
来源: https://blog.csdn.net/qq_40008772/article/details/120221336