算法初赛第十三题

题目描述

240. 搜索二维矩阵 II
     编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target 。该矩阵具有以下特性：

每行的元素从左到右升序排列。
每列的元素从上到下升序排列。

示例 1：

输入：matrix = [[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
示例 2：

输入：matrix = [[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 = 20
输出：false

解法一

解题思路

蛮力法
遍历整个数组,如果target in matrix返回True
否则返回False

python代码

def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        for i in matrix:
            if target in i:
                return True
        return False

运行结果

解法二

解题思路

从矩阵右上角开始查找，如果target大，那么向下查找，如果target小就向左查找
如：matrix矩阵中查找target=9
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
初始·i=0,j=4,matrix[i][j]=15>9,那么要向左查找，即j = j-1
matrix[i][j]=11>9,还是向左查找，j = j-1
matrix[i][j]=7<9,这时向下查找，i = i+1
matrix[i][j]=8<9,还是向下查找，i = i+1
matrix[i][j]=9==9，停止查找返回True

python代码

def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
        n,m = len(matrix)-1,len(matrix[0])-1
        i,j=0,m
        while i<=n and j>=0:
            if matrix[i][j]<target:
                i += 1
            elif matrix[i][j]==target:
                return True
            else:
                j -= 1

运行结果