在二维数组中搜索单个元素- Scala

def searchElement(matrix: Array[Array[Int]], target: Int): Boolean = {
  val rows = matrix.length
  val cols = matrix(0).length

  var row = 0
  var col = cols - 1

  while (row < rows && col >= 0) {
    if (matrix(row)(col) == target) {
      return true
    } else if (matrix(row)(col) > target) {
      col -= 1
    } else {
      row += 1
    }
  }

  false
}

// 示例用法
val matrix = Array(
  Array(1, 2, 3),
  Array(4, 5, 6),
  Array(7, 8, 9)
)

val target = 5
val result = searchElement(matrix, target)
println(s"元素 $target 是否存在于二维数组中: $result")