search-a-2d-matrix

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值