题目：给定一个有N*M的整型矩阵matrix和一个整数K， matrix的每一行和每一列都是排好序的。 实现一个函数， 判断K是否在matrix中。
例如：
0 1 2 5
2 3 4 7
4 4 4 8
5 7 7 9
如果K为7， 返回true； 如果K为6， 返回false。
要求：
时间复杂度为O(N+M)， 额外空间复杂度为O(1)。
public static boolean isContains(int[][] matrix, int K) {
int row = 0;
int col = matrix[0].length - 1;
while (row < matrix.length && col > -1) {
if (matrix[row][col] == K) {
return true;
} else if (matrix[row][col] > K) {
col--;
} else {
row++;
}
}
return false;
}
网友评论