Leetcode-240. 搜索二维矩阵 II

使用二分查找解决搜索二维矩阵问题

最新推荐文章于 2024-04-14 06:12:46 发布

原创最新推荐文章于 2024-04-14 06:12:46 发布 · 223 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

Leetcode（数学）专栏收录该内容

9 篇文章

订阅专栏

这篇博客讨论了一种高效算法，用于在给定的有序二维矩阵中搜索目标值。算法从矩阵的右上角开始，根据目标值与当前元素的比较，动态调整搜索方向。如果找到目标值，则返回true，否则返回false。这种方法适用于矩阵元素按行和列升序排列的情况。

链接

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;
    }
};