题目
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
输入:7,[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
输出:True
一、思路
[
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]
]
由于数组行和列都是递增的,利用这个特性,我们每次都从矩阵的左下角进行比较。假设输入为target,会有两种情况发生:
- 情况1 如果左下角元素比target大,则矩阵的最后一行都比target大,待比较矩阵可减少一行
- 情况2 假如左下角元素比target小,则矩阵的最左边一列都比target小,待比较矩阵可减少一列
二、代码
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
row=len(array)
col=len(array[0])
if row==0 or col==0:return False
i=0
j=col-1
while i<row:
if j==-1:break
while j>=0:
if array[i][j]==target:
return True
elif array[i][j]>target:
j=j-1
break
else:
i=i+1
break
if j==-1 or i==row:return False
总结
多利用给定的条件,发现数据的规律,减少计算量。