Search a 2D Matrix
[ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ]
Given
> target = 3
return
> truedef searchMatrix(matrix: Array[Array[Int]], target: Int): Boolean = {
if (matrix.isEmpty) false
else {
val (m, n) = (matrix.length, matrix(0).length)
val length = m * n
def find(start: Int, end: Int): Boolean = {
if (end < start) false
else {
val mid = (start + end) / 2
matrix(mid / n)(mid % n) match {
case num if num == target => true
case num if num < target => find(mid + 1, end)
case num if num > target => find(start, mid - 1)
}
}
}
find(0, length - 1)
}
}
val mat = Array(
Array(1, 3, 5, 7),
Array(10, 11, 16, 20),
Array(23, 30, 34, 50)
)
searchMatrix(mat, 11)
searchMatrix(mat, 99)Last updated