寫在開頭的話:
當想寫這個的時候,發現已經有人做了這個工作了,詳情請見Python基礎算法/劍指offer,然而依然決定自己寫下這個系列,作爲算法部分的鞏固和提高。在自己寫完後會借鑑Python基礎算法/劍指offer的代碼,如有部分重複,還請見諒。
# -*- coding: UTF-8 -*-.
'''
在一個二維數組中,每一行都按照從左到右嚴格遞增的順序排序
每一列都按照從上到下嚴格遞增的順序排序。
請完成一個函數,輸入這樣的一個二維數組和一個整數,判斷數組中是否含有該整數。
'''
'''
查找方式從右上角開始查找
如果當前元素大於target, 左移一位繼續查找
如果當前元素小於target, 下移一位繼續查找
'''
def solve(matrix, findNumber):
'Function to solve problem.'
if not isinstance(matrix, list):
print "invalid list"
return
elif not matrix:
print "empty list"
return
if not isinstance(findNumber, int):
print "invalid int"
return
flagFind = False
row, column = len(matrix), len(matrix[0])
x, y = 0, column - 1
while((x < row) & (y >= 0)):
if matrix[x][y] == findNumber:
flagFind = True
break
elif matrix[x][y] < findNumber:
x += 1
elif matrix[x][y] > findNumber:
y -= 1
if flagFind:
print 'find the number %d, first in row %d, column %d.' % (findNumber, x, y)
else:
print "can't find the number %d." % findNumber
def test():
'Test the program.'
matrix = [[1, 2, 8, 9], [2, 4, 9, 12], [4, 7, 10, 13], [6, 8, 11, 15]]
findNumber = 7
solve(matrix, findNumber)# find the number 7, in row 2, column 1
matrix = [[1, 2, 8, 9], [2, 4, 9, 12], [4, 7, 10, 13], [6, 8, 11, 15]]
findNumber = 5
solve(matrix, findNumber)# can't find the number 5
matrix = 3
findNumber = 4
solve(matrix, findNumber)# invalid list
# def input():
# 'Input by hand'
# # 輸入單個數字
# x = int(raw_input())
# # 輸入兩個數字
# [n, m] = map(int, raw_input().strip().split(' '))
# # 輸入一串數字
# a = map(int, raw_input().strip().split(' '))
# # 輸入字符串
# b = list(raw_input())
# solve(x, n, m, a, b)
if __name__ == "__main__":
#input()
test()