有序矩阵查找

来源:互联网 发布:mac装win7加载驱动程序 编辑:程序博客网 时间:2024/05/19 00:16

现在有一个行和列都排好序的矩阵,请设计一个高效算法,快速查找矩阵中是否含有值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 Falseif __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))
原创粉丝点击