二维数组中查找——python版本

题目描述

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

 

常规想法：

class Solution:

    # array 最小花费时间应该是先确定在哪一行搜索，然后再局部调优；发现从二维数组的左下角最合适搜索

        # 搜索最下面的一行，判断如果target大于第一个数，则列索引+1；如果target小于这个数，则行索引-1；如果等于,则返回true；跳出循环后，则返回false

        w_length=len(array[0])#得到列数

        i=h_length-1 #列索引

        while i>=0 and j<w_length:

            if target>array[i][j]:# <span style="" font-size:18.0pt;"="">如果target大于第一个数，则列索引+1；如果target小于这个数，则行索引-1；如果等于,则返回true

                j=j+1 

            elif target<array[i][j]:

                i=i-1

            else:

               return 1

        return 0

二分查找思路：每一行进行二分查找：

# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
# 最小花费时间应该是先确定在哪一行搜索，然后再局部调优；发现从二维数组的左下角最合适搜索
# 搜索最下面的一行，判断如果target大于第一个数，则列索引+1；如果target小于这个数，则行索引-1；如果等于,则返回true；跳出循环后，则返回false
w_length = len(array[0]) # 得到行数
i = 0
while i < w_length:
if self.bin_search(array[i], target):
return 1
return 0

def bin_search(self, array, val):
# 二分查找，确定底部、顶部和中间索引，如果中间值等于val，返回true；如果中间值在val上面，则顶部往下移动，索引值为中间索引-1；如果中间值在val下面，则底部往上移动，索引值为中间索引+1；止到底部和顶部重叠

bottom = 0 # 底部
top = len(array) - 1 # 顶部

while bottom <= top:
mid = (top + bottom) // 2
if array[mid] > val:
top = mid - 1
elif array[mid] < val:
bottom = mid + 1
else:
return 1
return 0