牛客网剑指offer第一题,二维数组的查找,原地址在这里https://www.nowcoder.com/ta/coding-interviews
想要联练习的小伙伴可以去练习练习。
题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
这道题对python来说,只需要target in array就可以了,但是逻辑能力和数据结构能力得不到提示。下面记录一下思路,截图取自剑指offer这本书,还是挺推荐购买正版的。
假设数组是[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]],需要发现7这个数字,先从右上角9开始,如果9大于Target,则不考虑9这一列;如果9小于target,则不考虑9这一行,如下截图所示
下面上代码,按照逻辑思路来写的,可能空间以及变量不是利用的最好
# -*- coding:utf-8 -*-
'''
#========================================
剑指offer二维数组查找
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
#========================================
'''
class Solution:
# array 二维列表
def Find(self, target, array):
row = len(array) #返回数组的行数
col = len(array[0]) #返回数组的列数
row_i = 0
col_i = col-1
result = False
while(col_i >= 0 and row_i < row):
if array[row_i][col_i] > target: #右上角大于目标,列减一
col_i = col_i - 1
elif array[row_i][col_i] < target: #右上角小于目标,行加一
row_i = row_i + 1
else:
result = True
break
return result
# write code here
while True:
try:
S=Solution()
# 字符串转为list
L=list(eval(raw_input()))
array=L[1]
target=L[0]
print(S.Find(target, array))
except:
break
然而对python来说,这道题其实很简单,python提供的强大的内置函数,下面是牛客网 网友的解答
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
n=len(array)
flag='false'
for i in range(n):
if target in array[i]:
flag='true';
break
return flag
while True:
try:
S=Solution()
# 字符串转为list
L=list(eval(raw_input()))
array=L[1]
target=L[0]
print(S.Find(target, array))
except:
break