【Killua笔试面试题整理】有序矩阵查找值

最新推荐文章于 2012-07-05 12:09:56 发布

原创最新推荐文章于 2012-07-05 12:09:56 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#面试 #matrix #python #c

☆IT笔试面试题整理☆ 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种在递增排序的二维数组中查找特定整数的算法。通过递归结合二分查找的方式，有效地缩小搜索范围，实现了高效查找。并提供了Python实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【试题描述】

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。（PS：数组的不一定是n*n的矩阵）

【试题来源】未知

【试题分析】

总体思路就是使用递归+二分查找的方法，具体过程如下所示：

设二分查找的中间点为(m_x,m_y)，其中

m_x = (s_x +e_x) / 2

m_y = (s_y +e_y) / 2

【源代码Python】

#!/usr/bin/env python

def find_matrix(mat, s_x, s_y, e_x, e_y, key):
    """
    mat:传入的矩阵
    (s_x, s_y):矩阵左上角坐标值
    (e_x, e_y):矩阵右下角坐标值
    """
    #明确key不在矩阵中的时候
    if mat[s_x][s_y] > key or mat[e_x][e_y] < key:
        return False

    #当矩阵规模为1*1的时候
    if s_x == e_x and s_y == e_y:
        if mat[s_x][s_y] == key:
            print("(%s, %s)" % (s_x, s_y))
            return True
        else:
            return False

    #中间点坐标值
    m_x = int((s_x + e_x) / 2)
    m_y = int((s_y + e_y) / 2)
    if mat[m_x][m_y] == key:
        print("(%s, %s)" % (s_x, s_y))
        return True
    elif mat[m_x][m_y] > key:
        #查询左上角,右上角,左下角矩阵
        return find_matrix(mat,s_x, s_y, m_x, m_y, key) \
               or find_matrix(mat,s_x, m_y + 1, m_x - 1, e_y, key) \
               or find_matrix(mat,m_x + 1, s_y, e_x, m_y - 1, key)
    elif mat[m_x][m_y] < key:
        #查询左下角,右上角,右下角矩阵
        return find_matrix(mat,m_x + 1, s_y, e_x, m_y, key) \
               or find_matrix(mat,s_x, m_y + 1, m_x, e_y, key) \
               or find_matrix(mat,m_x + 1, m_y + 1, e_x, e_y, key)

def main():
    matrix = [[1, 2, 3 , 4, 5],
              [6, 7, 8 , 9, 10],
              [11, 12, 13, 14, 15],
              [16, 17, 18, 19, 20],
              [21, 22, 23, 24, 25],
              [26, 27, 28, 29, 30],
              [31, 32, 33, 34, 35]]
    key = 24
    if find_matrix(matrix, 0, 0, len(matrix) - 1, len(matrix[0]) - 1, key):
        print("Found!")
    else:
        print("Not Found!")

if __name__ == '__main__':
    main()

【参考资料】

http://topic.youkuaiyun.com/u/20111214/10/d09797c3-d1ce-4249-b1e5-8b693b4c85f8.html

http://justjavac.iteye.com/blog/1310178

http://nubnub.blog.163.com/blog/static/169186347201192411857362/