剑指offer刷题 二维数组中的查找

题目

时间限制：1秒 空间限制：32768K 热度指数：1416236
本题知识点： 查找 数组
题目描述：
在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

解题

首先由二维数组的单调性特殊，所以肯定放弃暴力查找；由于是二维数组，所以二分不太适用，想到的是从左下角开始查找，因为左下角的数是该行最小的，又是该列最大的，比较有特点，可以据此向右上角过度，逐个查找。

具体思路

初始位置：左下角元素
终止位置：右上角元素

while 是否到达右上角元素：
	如果 行数<0 or 列数溢出：
		返回 False
	如果 target<当前元素：
		行数-1
	如果 target>当前元素：
		列数+1
	否则：
		target=当前元素
		返回True

代码

import numpy as np

class Solution:
    def find(self,target,array):
        m=len(array)-1
        n=0
        while m!= 0 or n!=len(array[0])-1:
            if m<0 or n>len(array[0])-1:
                return False
            elif target < array[m][n]:
                m-=1
                continue
            elif target > array[m][n]:
                n+=1
                continue
            else:
                return True

if __name__ == '__main__':
    target=int(input())
    n = int(input())
    a = []
    for _ in range(n):
        a.append(list(map(int, input().rstrip().split())))
    array=np.array(a)
    s=Solution()
    print(s.find(target, array))

其需要重视的是，由输入的二维数组转换的方法
rstrip()删除string末尾的指定字符，默认为空格