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

题目

给定一个有N*M的整型矩阵matrix和一个整数K，matrix的每一行和每一列都是排好序的（从左到右递增，从上到下递增）。实现一个函数，判断K是否在matrix中。

代码实现

public boolean isContains(int[][] matrix, int K) {
    //从矩阵最右上角的数开始寻找
    int row = 0;
    int col = matrix.length - 1;

    while (row < matrix.length && col > -1) {
        if (matrix[row][col] == K) {
            return true;
        } else if (matrix[row][col] < K) {
            row++;
        } else {
            col--;
        }
    }

    return false;
}