http://www.lintcode.com/zh-cn/problem/search-a-2d-matrix/
http://www.jiuzhang.com/solutions/search-a-2d-matrix/
描述:
写出一个高效的算法来搜索 m × n矩阵中的值。
这个矩阵具有以下特性:
每行中的整数从左到右是排序的。
每行的第一个数大于上一行的最后一个整数。
样例
考虑下列矩阵:
[
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
给出 target = 3,返回 true
这道题标答中的答案比较难想到,我按照我的代码来解释。
这道题的矩阵有一个性质,就是 每行中的第一个数大于上一行的最后一个整数 由此我们可以想到,先确定target在哪行,这道题就迎刃而解了,化归成了我们做过的题。
我写的这个代码比较粗糙,有足够多可以优化的地方,而且用的是我自己的写法,大家主要理解思想,不要去抠太多和模板不同的地方。
class Solution:
"""
@param matrix, a list of lists of integers
@param target, an integer
@return a boolean, indicate whether matrix contains target
"""
def searchMatrix(self, matrix, target):
if matrix == None or matrix.__len__() == 0: # 老规矩,先判断空
return False
start = 0
end = matrix.__len__() - 1 # 等价于 len(matrix) - 1,这里要理解一下,二维数组其实就是元素为数组的数组
while start + 1 < end: # 经典模板
mid = (start + end) / 2
if matrix[mid][0] >= target: # 这里我保证了target要么在mid之前,要么等于mid[0](就是在mid中)
end = mid
else:
start = mid
row = 0 # 循环结束后end一定等于start +1 所以我们只要判断一下再哪行中就可以了
if matrix[end][0] > target: # 第一种情况,不在end中,那就一定在或者不在start中(意思是有可能target不存在)
row = start
if matrix[end][0] <= target: # 第二种情况,在end中
row = end
start = 0 # 后面就是经典的二分法寻找位置了,就不多解释了
end = matrix[row].__len__() - 1
while start + 1 < end:
mid = (start + end) / 2
if matrix[row][mid] >= target:
end = mid
else:
start = mid
if matrix[row][end] == target or matrix[row][start] == target:
return True
return False