题目描述
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
常规想法:
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