题目
给定一个元素为非负整数的二维数组matrix, 每行和每列都是从小到大有序的。
再给定一个非负整数aim, 请判断aim是否在matrix中。
实现
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.Stack;
public class Main {
public static boolean find(int[][] m, int k) {
int row = 0;
int col = m[0].length - 1;
while (row != m.length && col != -1) {
if (m[row][col] > k) {
col--;
} else if (m[row][col] < k) {
row++;
} else {
return true;
}
}
return false;
}
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
int n = in.nextInt();
int m = in.nextInt();
int[][] matrix = new int[n][m];
int k = in.nextInt();
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
matrix[i][j] = in.nextInt();
}
}
System.out.println(find(matrix, k) ? "Yes" : "No");
break;
}
in.close();
}
}
文章评论