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

题目

给定一个元素为非负整数的二维数组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();
	}
}