二维数组中的查找(一)

# -*-coding:utf-8 -*-
'''
题目
    在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，
    每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，
    判断数组中是否含有该整数。
思路
    首先选取数组中左上角的数字。如果该数字等于要查找的数字，查找过程结束；
    如果该数字大于要查找的数组，剔除这个数字所在的列；如果该数字小于要查找的数组，
    剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，
    则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，
    直到找到要查找的数字，或者查找范围为空。
'''

"""
Author:
    Yao Zhian
Blog:
    https://cuijiahua.com/blog/2017/11/basis_1.html
Modify:
    2020-01-18
"""


def loadSimpData():
    """
    创建初始牌数组元素
    Parameters:
        无
    Returns:
        array - 初始二维数据
    """
    array = [
        [1, 2, 8, 9],
        [2, 4, 9, 12],
        [4, 7, 10, 13],
        [6, 8, 11, 15]
    ]

    return array

class Solution:
    # array 二维列表
    def Find(self, target, array):
        m = len(array)
        n = len(array[0])
        pos_x = m - 1 # 初始位置在左下角位置
        pos_y = 0
        while (pos_x >= 0)&(pos_y < n): # 超出边界就跳出
            if array[pos_x][pos_y] == target: # 如果当前位置数字等于要查找的数字，查找过程结束，返回True
                return True
            elif array[pos_x][pos_y] < target: # 如果当前位置数字小于要查找的数字，过滤掉一列
                pos_y = pos_y + 1
            else:                              # 如果当前位置数字大于要查找的数字，过滤掉一行
                pos_x = pos_x - 1
        return False
        # write code here


if __name__ == '__main__':
    array = loadSimpData()
    # print(num_color)
    S = Solution()
    print(S.Find(7, array))
二维数组中的查找(一)

推荐阅读
