leetcode 刷题之路 52 Search a 2D Matrix

Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:

• Integers in each row are sorted from left to right.
• The first integer of each row is greater than the last integer of the previous row.

For example,

Consider the following matrix:

[
  [1,   3,  5,  7],
  [10, 11, 16, 20],
  [23, 30, 34, 50]
]

Given target = 3, return true.

给定一个特殊矩阵，判断其中是否包含目标数字，有返回true，否则返回false。

这个矩阵的定义是这个样子的，每行的元素从左到右递增，每一行的第一个元素比上一行的最后一个元素大。

杨氏矩阵的定义是这个样子的，每行的元素从左到右递增，每列的元素从上到下递增。

可见，这个矩阵是杨氏矩阵，我们可以使用杨氏矩阵的查找方法进行查找，从矩阵的右上角开始，比较右上角数据和目标数据大小，若目标数据大于右上角数据，则最上面一行都小于目标数据，排除掉一行；若目标数据小于右上角数据，则最右边一列都大于目标数据，排除掉一列。这样子一直下去可以利以O(m+n)时间复杂度完成查找操作。

AC code：

class Solution {
public:
    bool searchMatrix(vector<vector<int> > &matrix, int target) 
    {
        int rowNum=matrix.size(),j=matrix[0].size()-1,i=0;
        while(i<rowNum&&j>=0)
        {
            if(matrix[i][j]==target)
                return true;
            if(target>matrix[i][j]) //大于行尾元素，本行排除
                i++;
            else                    //小于列首元素，本行排除
                j--;
        }
        return false;
    }
};

但是，这个矩阵比杨氏矩阵要更特殊，我们从每一行的行首遍历到行末再折回到到下一行行首一直遍历下去，可以得到一个升序的序列，利用这个条件，可以使用二分查找法解决这个问题。

AC code：

class Solution {
public:
    bool searchMatrix(vector<vector<int> > &matrix, int target) 
    {
        if(matrix.empty())
            return false;
        int begin=0,rowNum=matrix.size(),clmnNum=matrix[0].size(),end=rowNum*clmnNum-1;
        int mid;
        while(begin<=end)
        {
            mid=(begin+end)/2;
            if(matrix[mid/clmnNum][mid-(mid/clmnNum)*clmnNum]==target)
                return true;
            if(matrix[mid/clmnNum][mid-(mid/clmnNum)*clmnNum]>target)
                end=mid-1;
            else
                begin=mid+1;
        }
        return false;
    }
};

时间复杂度O(log(m*n))，题目条件利用的更充分，比第一种方法时间效率要高。