Leetcode-240. 搜索二维矩阵 II

链接

240. 搜索二维矩阵 II

题目

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

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

示例

示例 1：
输入：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 

示例 2：
输入：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 = 20
输出：false 

说明

• m == matrix.length
• n == matrix[i].length
• 1 <= n, m <= 300
• -10e9 <= matrix[i][j] <= 10e9
• 每行的所有元素从左到右升序排列
• 每列的所有元素从上到下升序排列
• -10e9 <= target <= 10e9

思路

这题很容易想到二分法的思路，但这不是最优解。

我们可以从矩阵的右上方往左下方搜索，如果当前的值＞target，由于每列的元素从上到下升序排列，必然当前列的元素都要更大，因此列数-1，如果当前的值＜target，而又由于每行的元素从左到右升序排列，故将行数+1.

C++ Code

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        //从右上角开始检查，如果target>该值 行数+1 如果小于 列数-1
        int m=0, n=matrix[0].size()-1;
        while(m<matrix.size() && n>=0)
        {
            if(matrix[m][n]==target) return true;
            else if(matrix[m][n]<target) m++;
            else n--;
        }
        return false;
    }
};