题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解答如下:
环境:python 2.7.3
1 2 3 4
3 4 5 6
4 6 8 10
9 11 13 15
本题当然可以使用双重for循环的暴力解法,挨个遍历数组元素并和目标整数做对比,但是那样的时间复杂的是O(mn),太大了。
其实,本题的正确思路是从二维矩阵的右上角往下对比,比如上面的例子,我去找8,4先和8对比4<8,6<8,10>8,然后再往10这一行的左侧对比,8==8,对比结束。
其实,每一行最右侧的数是这一行的最大值,如果它比目标整数小,那么这一行都比目标小,在上面的例子中,当10大于8时,8也一定不在前两行,因为8若在前两行的最右侧那程序早就结束了,所以如果存在8,那一定在10的左边。
最后,要有8就返回True,没有则返回False。
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
# 1 2 3 4
# 3 4 5 6
# 4 6 8 10
# 9 11 13 15
#if target == 8
row_count = len(array)
column_count = len(array[0])
i = 0
j = column_count - 1
while i < row_count and j >= 0:
value = array[i][j]
if value == target:
return True
elif value > target:
j -= 1
else:
i += 1
return False