题目:
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题代码一:二层遍历,时间复杂度O(mn)
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
m,n=len(array),len(array[0])
for i in range(m):
for j in range(n):
if array[i][j]==target:
return True
return False
解题代码二:先与左下角或者右上角的值先比较。若target>右上角的值,则可以先排除第一行,剩余三行;若target<右上角,则排除最后一列,剩余三列
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
m,n=len(array),len(array[0])
row=0
col=n-1 #右上角的值
while(row<m and col>=0):
if array[row][col]<target:
row+=1
elif array[row][col]>target:
col-=1
else:
return True
return False
时间复杂度:O(n+m)。访问到的下标的行最多增加 n 次,列最多减少 m 次,因此循环体最多执行 n + m 次。
空间复杂度:O(1)。