本文始发于个人公众号:TechFlow,原创不易,求个关注
今天是LeetCode专题43篇文章,我们今天来看一下LeetCode当中的74题,搜索二维矩阵,search 2D Matrix。
这题的官方难度是Medium,通过率是36%,和之前的题目不同,这题的点赞比非常高,1604个赞,154个反对。可见这题的质量还是很高的,事实上也的确如此,这题非常有意思。
题意
这题的题意也很简单,给定一个二维的数组matrix和一个整数target,这个数组当中的每一行和每一列都是递增的,并且还满足每一行的第一个元素大于上一行的最后一个元素。要求我们返回一个bool变量,代表这个target是否在数组当中。
也就是说这个是一个典型的判断元素存在的问题,我们下面来看看两个样例:
Input:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
Output: true
Input:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
Output: false
题解
这题刚拿到手可能会有些蒙,我们当然很容易可以看出来这是一个二分的问题,但是我们之前做的二分都是在一个一维的数组上,现在的数据是二维的,我们怎么二分呢?
我们仔细阅读一下题意,再观察一下样例,很容易发现,如果一个二维数组满足每一行和每一列都有序,并且保证每一行的第一个元素大于上一行的最后一个元素,那么如果我们把这个二维数组reshape到一维,它依然是有序的。
比如说有这样一个二维数组:
[[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
它reshape成一维之后会变成这样:
[1, 2, 3, 4, 5, 6, 7, 8, 9]
reshape是numpy当中的说法,也可以简单理解成把每一行串在一起。所以这题最简单的做法就是把矩阵降维,变成一位的数组之后再通过二分法来判断元素是否存在。如果偷懒的话可以用numpy来reshape,如果不会numpy的话,可以看下我之前关于numpy的教程,也可以自己用循环来处理。
reshape之后就是简单的二分了,完全没有任何难度:
class Solution:
def searchMatrix(self, matrix: List[List[int]], target: int) ->