《剑指offer》003-二维数组中的查找（更接近笔试--涉及到输入函数）

题目：在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排列。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该函数。

输入：

　　输入可能包含多个测试样例，对于每个测试案例， 

　　输入的第一行为两个整数m和n(1<=m,n<=1000)：代表将要输入的矩阵的行数和列数。 

　　输入的第二行包括一个整数t(1<=t<=1000000)：代表要查找的数字。 

　　接下来的m行，每行有n个数，代表题目所给出的m行n列的矩阵(矩阵如题目描述所示，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

输出：

　　对应每个测试案例， 

　　输出”Yes”代表在二维数组中找到了数字t。 

　　输出”No”代表在二维数组中没有找到数字t。　

样例输入：

3 3 

5 

1 2 3 

4 5 6 

7 8 9 

3 3 

1 

2 3 4 

5 6 7 

8 9 10 

3 3 

12 

2 3 4 

5 6 7 

8 9 10

样例输出：

Yes 

No 

No

解题思路：首先选取数组中右上角的数字，如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数字，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

import java.util.*;

public class T003{
public static void main(String[] args){
Scanner in = new Scanner(System.in);
int rows = 0, cols = 0, number = 0;
rows = in.nextInt();
cols = in.nextInt();
number = in.nextInt();
int a[][] = new int[rows][cols];
for(int i = 0; i < rows; j++){
for(int j = 0; j < cols; j++){
a[i][j] = in.nextInt();
}
}
if(find(a, rows, cols, number))
System.out.println("Yes");
else System.out.println("No");
}
public static boolean find(int a[][], int rows, int cols, int f) {
int row = 0, col = cols - 1;
while (row < rows && col >= 0) {
if (a[row][col] == f) {
return true;
} else if (a[row][col] > f) {
--col;
} else {
++row;
}
}
return false;
}
}