《剑指offer》003-二维数组中的查找

《剑指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;
     }
}