牛客刷题-剑指offer(1)
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路:
1. 找对角线,这样做比较复杂,因为不一定是方阵。烂思路。
2. 右上角,如果target大于右上角,删除第一行;如果target小于右上角,删除最后一列,时间复杂度为O(n)
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
row, col = 0, len(array[0]) - 1
height = len(array)
while (row < height and col > -1):
if target > array[row][col]:
row += 1
elif target < array[row][col]:
col -= 1
else:
return True
return False
# --------test code -----------
test = Solution()
result = test.Find(16, [[]])
print(result)