《剑指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()