[LintCode]28.搜索二维矩阵

来源:互联网 发布:怎样加入淘宝 编辑:程序博客网 时间:2024/05/22 08:29

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

这个矩阵具有以下特性:每行中的整数从左到右是排序的。

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

样例

考虑下列矩阵:

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

给出 target = 3,返回 true

思路:从左下角或右上角开始搜索均可,每次判断大小换行或者换列移动。比较matrix[row][col]与target的关系。

class Solution {public:    /**     * @param 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) {        if(matrix.empty() || matrix[0].empty())          {            return false;         }         int row=0;          int col=matrix[0].size()-1;          while(row<matrix.size() && col>-1){              if(matrix[row][col]==target){                  return true;            }else if(matrix[row][col] > target){                  col--;              }else{                  row++;              }          }        return false;    }};