《劍指offer》Python語言 面試題3:二維數組中的查找

    寫在開頭的話:

    當想寫這個的時候,發現已經有人做了這個工作了,詳情請見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()


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章