[LeetCode][C++]搜索二维矩阵 II

搜索二维矩阵 II

编写一个高效的算法来搜索 m x n 矩阵 matrix 中的一个目标值 target。该矩阵具有以下特性:

  • 每行的元素从左到右升序排列。
  • 每列的元素从上到下升序排列。

示例:
现有矩阵 matrix 如下:
[
\quad [1, 4, 7, 11, 15],
\quad [2, 5, 8, 12, 19],
\quad [3, 6, 9, 16, 22],
\quad [10, 13, 14, 17, 24],
\quad [18, 21, 23, 26, 30]
]
给定 target = 5,返回 true。
给定 target = 20,返回 false。


思路:
从矩阵的负对角(左下角,右上角)开始搜索,直至边界:
例如从左下角开始搜索,

  • 若当前元素 > t a r g e t target target,则往上走。
  • 若当前元素 < t a r g e t target target,则往右走。
  • 若相等,则返回true。
    [LeetCode][C++]搜索二维矩阵 II_第1张图片

代码:

bool searchMatrix(vector<vector<int>>& matrix, int target) {
        //矩阵为空的判断,否则通过不了,测试例子有空矩阵。
        if(matrix.size()==0 || matrix[0].size()==0) 
            return false;                
        int row=matrix.size();
        int col=matrix[0].size();
         //从负对角线开始搜索,副对角线两头均可以作为起点              
        int i=row-1,j=0;    
        while(i>=0 && j<col)     
        {
            if(matrix[i][j]==target)
            {
                return true;
            }else if(matrix[i][j]<target)
            {
                j++;
            }else if(matrix[i][j]>target)
            {
                i--;
            }          
        }
        return false;
    }

结果:
[LeetCode][C++]搜索二维矩阵 II_第2张图片
扩展:
你可以试试从右上角开始搜索。

你可能感兴趣的:(刷题)