LeetCode——240.搜索二维矩阵 II(Search a 2D Matrix II)[中等]——分析及代码（C++、Java）

一、题目

编写一个高效的算法来搜索 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
• -10^9 <= matrix[i][j] <= 10^9
• 每行的所有元素从左到右升序排列
• 每列的所有元素从上到下升序排列
• -10^9 <= target <= 10^9

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-a-2d-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 从左下或右上搜索/Z字形查找

（1）思路

若按下标从左上角搜索，由于向下、向右数值均单调递增，导致无法准确判断搜索方向；

若从左下角搜索，则可以准确判断搜索方向：

• 当前值 > target，搜索上方数值；
• 当前值 < target，搜索右方数值；

可以不断迭代，直至搜索到目标 target （true） 或 超出矩阵范围（false）。

（2）代码（C++）

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int i = matrix.size() - 1, j = 0;
        while (i > -1 && j < matrix[0].size()) {
            if (matrix[i][j] > target)
                i--;
            else if (matrix[i][j] < target)
                j++;
            else   
                return true;
        }
        return false;
    }
};

（3）结果（C++）

执行用时 ：64 ms, 在所有 C++ 提交中击败了 94.18% 的用户；
内存消耗 ：12.6 MB, 在所有 C++ 提交中击败了 95.57% 的用户。

（4）代码（Java）

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int m = matrix.length, n = matrix[0].length;
        int i = m - 1, j = 0;//搜索指针的行、列坐标
        while (i >= 0 && j < n) {//指针在矩阵范围内时
            if (matrix[i][j] == target) {//找到目标值
                return true;
            } else if (matrix[i][j] < target) {//当前元素小于目标值，增大列坐标
                j++;
            } else {//当前元素大于目标值，减小行坐标
                i--;
            }
        }
        return false;//搜索完成后未在矩阵中发现目标值
    }
}

（5）结果（Java）

执行用时 ：5 ms，在所有 Java 提交中击败了 96.28% 的用户；
内存消耗 ：44 MB，在所有 Java 提交中击败了 39.76% 的用户。

2. 二分搜索

（1）思路

先在矩阵的 45 度对角线上二分搜索，直至发现 target（返回true）或找到 matrix[i][i] < target < matrix[i + 1][i + 1]。

将此时的矩形划分为 4 块讨论，左上部分的数值全部小于 target，右下部分的数值全部大于 target，可跳过。

针对左下部分和右上部分的矩阵进一步二分迭代，直至发现 target。

（2）代码

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        if (matrix.empty() || matrix[0].empty())
            return false;
        return search(matrix, target, 0, 0, matrix.size(), matrix[0].size());
    }

    bool search(vector<vector<int>>& matrix, int target, int li, int lj, int ri, int rj) {//参数为所搜索矩阵的左下、右上坐标
        if ((li > ri - 1) || (lj > rj - 1) || matrix[li][lj] > target || matrix[ri - 1][rj - 1] < target)
            return false;
        int li0 = li, lj0 = lj, ri0 = ri, rj0 = rj;     
        while (li < ri - 1 || lj < rj - 1) {//二分搜索对角线
            int mi = (li + ri) / 2, mj = (lj + rj) / 2;
            if (matrix[mi][mj] > target) {
                ri = mi;
                rj = mj;
            }
            else {
                li = mi;
                lj = mj;
            }
        }
        if (matrix[li][lj] == target)
            return true;
        return search(matrix, target, li + 1, lj0, ri0, lj + 1) || search(matrix, target, li0, lj + 1, li + 1, rj0);//递归搜索左下和右上矩阵
    }
};

（3）结果

执行用时 ：140 ms, 在所有 C++ 提交中击败了 31.13% 的用户；
内存消耗 ：13.1 MB, 在所有 C++ 提交中击败了 5.06% 的用户。

三、其他

暂无。