leetcode刷题笔记 240题 搜索二维矩阵 II

leetcode刷题笔记 240题 搜索二维矩阵 II

源地址：240. 搜索二维矩阵 II

问题描述：

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

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

现有矩阵 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。

给定 target = 20，返回 false。

/*
从右上角开始遍历，若其大于target，排除当前列，右移
否则排除当前行，下移
直到找到target或抵达边界
*/
object Solution {
    def searchMatrix(matrix: Array[Array[Int]], target: Int): Boolean = {
        if (matrix.length == 0 || matrix(0).length == 0) return false
        
        val n = matrix.length
        val m = matrix(0).length
        var i = 0
        var j = m - 1 

        while (i < n && j >= 0) {
            val temp = matrix(i)(j)
            if (temp == target) return true
            else if (temp > target) j -= 1
            else i += 1
        }
        return false
    }
}