[Leetcode] 240. Search a 2D Matrix II

最新推荐文章于 2021-03-02 01:17:16 发布

原创最新推荐文章于 2021-03-02 01:17:16 发布 · 129 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

LeetCode 同时被 2 个专栏收录

70 篇文章

订阅专栏

#matrix

2 篇文章

订阅专栏

本文提供了一种解决LeetCode上Search a 2D Matrix II问题的有效算法。该算法利用了矩阵的特性，通过从逆对角线的两端开始搜索，实现了高效查找目标值的功能。适用于所有类型的二维矩阵搜索问题。

Problems:
https://leetcode.com/problems/search-a-2d-matrix-ii/

Solution:
鉴于matrix的特性，binary search已经不适用，为了迭代方便，从逆对角线的两端下手（任意选一端即可）

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if(matrix.length == 0 || matrix == null) {
            return false;
        }
        // 此处选的是逆对角线的下端
        // 由于matrix的特性，可以将搜索简化成两条线：大于target则向右搜索，小于target则向左搜索
        // 对角线两端不适用的原因在于，从起点开始target只存在大于/小于的情况
        int m = matrix.length, n = matrix[0].length;
        int curRow = m-1, curCol = 0;
        while (curRow >= 0 && curCol < n) {
            if(matrix[curRow][curCol] == target) {
                return true;
            } else if (matrix[curRow][curCol] < target) {
                curCol++;
            } else {
                curRow--;
            }
        }
        return false;
    }
}