# -*-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))