28. 搜索二维矩阵

提示

        LintCode中的相关算法题实现代码，可以在我的GitHub中下载。

题目需求

    

描述

            写出一个高效的算法来搜索 m × n矩阵中的值。

            这个矩阵具有以下特性：

                        每行中的整数从左到右是排序的。

                        每行的第一个数大于上一行的最后一个整数。

    样例

            考虑下列矩阵：

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

给出 target = 3，返回 true

挑战

O(log(n) + log(m)) 时间复杂度

解题思路    

       由于二维数组是从左到右递增，从上到下递增，我们可以这个可以考虑到：

        1.从二维数组的右上角判断，如果等于target，那么就直接返回。       

        2. 如果target小于右上角的元素，那么就判断下一列的第一个元素；

        3.否者就向下判断。

实现代码

class Solution {
public:
    /**
     * @param matrix: matrix, a list of lists of integers
     * @param target: An integer
     * @return: a boolean, indicate whether matrix contains target
     */
    bool searchMatrix(vector<vector<int>> &matrix, int target) {
        // write your code here
        if(matrix.size()==0) return false;
        int size=matrix.size();
        int size2=matrix[0].size();
        for(int i=size2-1;i>=0;i--)
        {
            if(matrix[0][i]==target)
            {
                return true;
            }
            else if(matrix[0][i]<target)
            {
                for(int j=1;j<size;j++)
                {
                    if(matrix[j][i]==target)
                    {
                        return true;
                    }
                    else if(matrix[j][i]>target) break;
                }
            }
        }
        return false;
    }
};