最近在刷牛客网上的《剑指 offer》,练一练自己的算法。有一些比较难的代码可能参考了牛客网的优质解答。
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
Python2 解答
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
row = len(array) # 计算有多少行
col = len(array[0]) # 计算有多少列
i = row - 1; j = 0 # 初始化行i;列j,从左下角开始寻找
while i >= 0 and j <= col - 1: # 循环条件
if target == array[i][j]:
return True
elif target < array[i][j]: # 若target小于当前元素,则往上找
i -= 1
else: # 若target大于当前元素,则往右找
j += 1
return False # 都没找到,输出