剑指Offer_编程题01:二维数组中的查找

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

首先选取数组中右上角的数字。如果该数字等于要查找的数字,查找过程结束;如果该数字大于要查找的数组,就向它前一列找,如果该数字小于要查找的数组,就向他下一行找。

就这么简单的一道题,之前都做过,再做一遍,花了一个小时,还到处是问题!

首先不能用for,因为这不是遍历,得用while循环,没符合条件时不能退出。

其次,我是这么做的。。一直报错数组越界:

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        # write code here
        row = len(array)
        col = len(array[0])

        if len(array) == 0:
            return False

        i = 0
        j = col - 1
        while (i < row and j >= 0):
            if target > array[i][j]:
                i += 1
            if target < array[i][j]:
                j -= 1
            if target == array[i][j]:
                return True

        return False

后来才发现,问题是三个if判断,并列是不对的,得用if else这样,每次循环只能进行一次操作,如果要是都用if,在一次循环里就可能会出现又加又减。。。好气。。。。

正确代码:

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        # write code here
        row = len(array)
        col = len(array[0])

        if len(array) == 0:
            return False

        i = 0
        j = col - 1
        while (i < row and j >= 0):
            if target == array[i][j]:
                return True
            elif target > array[i][j]:
                i += 1
            else:
                j -= 1

        return False

if __name__ == '__main__':
    a = [[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]]
    b = Solution()
    print(b.Find(14, a))

猜你喜欢

转载自blog.csdn.net/mengmengdajuanjuan/article/details/81706555