面试题3：二维数组中的查找

代码如下：

# coding:utf-8
"""
=========================================
File:SearchMatrix.py
Author:Nobita
Time:2019/12/18 20:46
E-main:364942727@qq.com
Description：面试题3：二堆数组中的查找
Software:PyCharm
=========================================
"""

'''
题目：
在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。
请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否有该整数。

输入:
matrix = [
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]
target_true = 3
输出: True
target_false = 13
输出: False
'''
class Solution:
    def searchMarix(self, matrix, target):
        num_row = len(matrix)
        num_col = len(matrix[0])
        if num_row > 0 and num_col > 0:
            i, j = 0, num_col - 1
            while i < num_row and j >= 0:
                if target == matrix[i][j]:
                    return True
                elif target < matrix[i][j]:
                    j -= 1
                else:
                    i += 1
        return False

if __name__ == '__main__':
    matrix = [
        [1, 3, 5, 7],
        [10, 11, 16, 20],
        [23, 30, 34, 50]
    ]
    target_true = 3
    target_false = 13
    print(Solution().searchMarix(matrix, target_true))
    print(Solution().searchMarix(matrix, target_false))

代码自测：

# -*- coding: utf-8 -*-
"""
@File:test_SearchMatrix.py    
@E-mail:364942727@qq.com
@Time:2019-12-19 8:24 
@Author:Nobita   
@Version:1.0   
@Desciption:对SearchMatrix函数进行单元测试
"""

import unittest
from .SearchMatrix import Solution


class TestSearchMatrix(unittest.TestCase):
    def setUp(self):
        global matrix
        matrix = [[1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50]]

    def test_01_MaxNum_in_Matrix(self):
        '''测试二维数组中的最大值,预期输出:True'''
        target = 50
        result = Solution().searchMarix(matrix, target)
        self.assertEqual(result, True, msg='test_01_测试失败！')

    def test_02_MinNum_in_Matrix(self):
        '''测试二维数组中的最小值,预期输出:True'''
        target = 1
        result = Solution().searchMarix(matrix, target)
        self.assertEqual(result, True, msg='test_02_测试失败！')

    def test_03_MidNum_in_Matrix(self):
        '''测试二维数组中的中间值,预期输出:True'''
        target = 11
        result = Solution().searchMarix(matrix, target)
        self.assertEqual(result, True, msg='test_03_测试失败！')

    def test_04_OverMaxNum_not_in_Matrix(self):
        '''测试二维数组中超过最大值的数,预期输出:False'''
        target = 51
        result = Solution().searchMarix(matrix, target)
        self.assertEqual(result, False, msg='test_04_测试失败！')

    def test_05_LessMinNum_not_in_Matrix(self):
        '''测试二维数组中小于最小值的数,预期输出:False'''
        target = 0
        result = Solution().searchMarix(matrix, target)
        self.assertEqual(result, False, msg='test_05_测试失败！')

    def test_06_MatrixIsNull(self):
        '''测试二维数组中空值矩阵,预期输出:False'''
        target = 1
        null_matrix = [[]]
        result = Solution().searchMarix(null_matrix, target)
        self.assertEqual(result, False, msg='test_06_测试失败！')


if __name__ == '__main__':
    unittest.main()