240. 搜索二维矩阵 II Java解法

剑指Offer 第二版 第4题

leetcode-cn.com/problems/se…

示例: 现有矩阵 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]

]

首先，不得不说这题和他的第一个版本的思想上是有点像的，但是又不是完全一样，因为第一版本中我们可以吧取值的范围缩小到一行当中，但是在这里是行不通的，因为这题的关键的是，最后答案的取值取值范围会呈现出一个矩形的形状分布。 这个题目的需要你从右上角的元素或者左下角的元素开始寻找。 如果目标比当前的元素大，那么就把这一行排除；如果目标值比当前的元素小，那么就把之一列给排除。 我把整个寻找的过程做了流程图，方便大家理解：

理清了思想，我们来看看具体的代码：

show the code.

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        if (matrix == null) {
            return false;
        } else if (matrix.length == 0) {
            return false;
        } else if (matrix[0].length == 0) {
            return false;
        }
        int len = matrix.length;
        int rowLen = matrix[0].length;
        int targetCol = rowLen - 1;

        int targetRow = 0;
        while (targetRow < len && targetCol >= 0) {
            if (matrix [targetRow][targetCol] < target) {
                targetRow++;
            } else if (matrix[targetRow][targetCol] > target) {
                targetCol--;
            } else {
                return true;
            }
        }
        return false;
    }
}

时间就是O（m+n），具体来看看效果，如下截图所示，还是不错的。