有序矩阵查找

最新推荐文章于 2022-01-14 22:02:48 发布

小旋锋

最新推荐文章于 2022-01-14 22:02:48 发布

阅读量630

点赞数

分类专栏：求职算法文章标签：排序算法

本文链接：https://blog.csdn.net/wwwdc1012/article/details/77532610

版权

求职算法专栏收录该内容

41 篇文章 0 订阅

订阅专栏

现在有一个行和列都排好序的矩阵，请设计一个高效算法，快速查找矩阵中是否含有值x。

给定一个int矩阵mat，同时给定矩阵大小nxm及待查找的数x，请返回一个bool值，代表矩阵中是否存在x。所有矩阵中数字及x均为int范围内整数。保证n和m均小于等于1000。

测试样例：
[[1,2,3],[4,5,6],[7,8,9]],3,3,10
返回：false

我的提交

# -*- coding:utf-8 -*-

class Finder:
    def findX(self, mat, n, m, x):
        # write code here
        i = 0
        j = m - 1
        while i < n and j >= 0:
            if mat[i][j] == x:
                return True
            if x > mat[i][j]:
                i += 1
            else:
                j -= 1
        return False

if __name__ == '__main__':
    f = Finder()
    print(f.findX([[109,204,260,270],[452,602,671,917],[1106,1343,1467,1585],[1627,1866,1948,1980],[2108,2164,2295,2577],[2681,2749,2915,3026],[3187,3250,3465,3518],[3562,3773,3966,4101]],8,4,2108))