在行列都排好序的矩阵中找指定的数

import scala.io.StdIn

object Main{
    def main(args: Array[String]): Unit = {
        val in = StdIn
        var params = in.readLine().split(" ")
        val n = params(0).toInt
        val m = params(1).toInt
        val k = params(2).toInt
        val arr = Array.ofDim[Int](n, m)
        for(i <- arr.indices){
            params = in.readLine().split(" ")
            for(j <- arr(0).indices)
                arr(i)(j) = params(j).toInt
        }
        println(search(arr, n, m, k))
    }
    
    def search(arr: Array[Array[Int]], n: Int, m: Int, k: Int): String = {
        var row = 0
        var col = m - 1
        while(row < n && col >= 0){
            if(arr(row)(col) > k){
                col -= 1
            }else if(arr(row)(col) < k){
                row += 1
            }else{
                return "Yes"
            }
        }
        "No"
    }
}