【程序员面试金典】面试题 10.09. 排序矩阵查找

题目描述

描述：给定M×N矩阵，每一行、每一列都按升序排列，请编写代码找出某元素。

示例:

现有矩阵 matrix 如下：

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]

给定 target = 5，返回 true。

给定 target = 20，返回 false。

解题思路

思路1：最直观的想法是，二维二分查找。选择从右上角出发：如果该值等于目标值，则返回true；如果该值小于目标值，则需要找更大的值，故向下走，row++；如果该值大于目标值，则需要找更小的值，故向左走，col–。

bool searchMatrix(vector<vector<int>>& matrix, int target) 
{
   int m=matrix.size();
   //考虑数组为空啊！注意位置啊！！如果m=0那么n就无法计算啊！！！
   if(m==0)
     return false;
   int n=matrix[0].size();
   if(n==0)
     return false;
   int row=0;
   int col=n-1;
   while(row<m&&col>=0)
   {
      if(matrix[row][col]==target)
        return true;
      //大于往左边走
      else if(matrix[row][col]>target)
        col--;
      //小于往下边走
      else if(matrix[row][col]<target)
        row++;
   }
   return false;
}

总结：行有序且列有序的二维矩阵也是可以使用二分查找的！！从左下角或者右上角出发均可以！！！注意调整方向！！！