剑指offer之二维数组中的查找（Python）

题目描述

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

思路1：利用数组的特点，从第一行开始，我们每次让数组的每行的最后一个和目标值进行比较。具体算法如下：

1、行序列  i 初始化为0，列序列  j 初始化为数组的列数；

2、写一个while循环判断： array[ i][ j] 和 target 的大小关系：

①array[ i][ j] < target， 说明这一行的数都比较目标值小，则意味着行序列需要增加；

②array[ i][ j] > target， 说明目标值可能就在这行，我们让列序列减小；

③array[ i][ j] = target， 那就说明找到了，返回True就好了

3、如果while循环都走完了，则只能说明这个数组中没有这个数，返回False

程序附上：

def function1(array,target):
    len_line = len(array) - 1
    len_column = len(array[0]) - 1
    i = 0
    j = len_column
    while i <= len_line and j >= 0:
        if array[i][j] < target:
            i += 1
        elif array[i][j] > target:
            j -= 1
        else:
            return True
    return False

思路2：利用Python自身特点，直接用for遍历每一行，然后用if判断是否在该列中。

1、如果找到了，返回True

2、循环结束后都还没有，则返回False

代码附上：

def function2(array,target):
    for line in array:
        if target in line:
            return True
        else:
            continue
    return False

最后，附上测试总代码：

# -*- coding: utf-8 -*-
"""
Created on Sat Oct 14 16:09:00 2017
@author: gb_xiao
Mail: mingliumengshao@163.com
"""
"""
题目描述：在一个二维数组中，每一行都按
照从左到右递增的顺序排序，每一列都按照
从上到下递增的顺序排序。请完成一个函数，
输入这样的一个二维数组和一个整数，判断
数组中是否含有该整数。
"""
def function1(array,target):
    len_line = len(array) - 1
    len_column = len(array[0]) - 1
    i = 0
    j = len_column
    while i <= len_line and j >= 0:
        if array[i][j] < target:
            i += 1
        elif array[i][j] > target:
            j -= 1
        else:
            return True
    return False

def function2(array,target):
    for line in array:
        if target in line:
            return True
        else:
            continue
    return False


def main():
    array = [[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]]
    target = 5
    print function1(array,target)
    print function2(array,target)
if __name__ == "__main__":
    main()