剑指offer-python

面试题4:二维数组的查找

2018-06-17  本文已影响0人  小歪与大白兔

题目描述

在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

解题思路

时间复杂度 O(n)的解题方法:

1.首先判断二维数组是否为空,不要遗忘掉

2.从二维数组的右上角array[0][len(array[0])-1]开始做比较,向左递减,向下递增;如果该数字大于target,则删除该数字所在列,如果该数字小于target,则删除该数字所在行。

# -*- coding:utf-8 -*-
class Solution:
    def Find(self, target, array):
        # write code here
        if len(array) == 0:
            return False
        row = 0
        col = len(array[0]) - 1
        #从右上角array[0][len(array[0])-1]开始做比较,向左递减,向下递增
        while row < len(array) and col >= 0:
            if array[row][col] == target:
                return True
            elif array[row][col] > target:
                col = col - 1
            else:
                row = row + 1
        return False
上一篇 下一篇

猜你喜欢

热点阅读