二维数组中的查找

题目来自剑指offer，如下图所示。

下图是解法。

package com.jeaven;
import java.util.Scanner;

/*在二维数组中查找数字*/
public class Problem4 {
    public static void main(String[] args) {
        int num = 0; //待查找的数字
        int row = 0;  //数组行数
        int col = 0;   //数组列数
        Scanner reader = new Scanner(System.in);
        System.out.println("请输入数组的行数：");
        if(reader.hasNextInt()) {
            row = reader.nextInt();
        }
        System.out.println("请输入数组的列数：");
        if(reader.hasNextInt()) {
            col = reader.nextInt();
        }
        int[][] arr = new int[row][col]; //初始化数组
        for(int i = 0; i < row; i++) {

            System.out.println("请输入第 " + (i+1) + " 行的数");
            for(int j = 0; j < col; j++) {
                if(reader.hasNextInt()) {
                    arr[i][j] = reader.nextInt();
                }
            }
        }
        System.out.println("请输入需要查找的数字: ");
        if(reader.hasNextInt()) {  //输入一个等待查找的数字
            num = reader.nextInt();
        }
        new Problem4_Solution1().solve(arr, num);
    }
}


/*解法：从数组的左下角或者右上角开始查找，下面从左下角开始查找
*依次缩小搜索范围，小于当前的数则减少下面一行，大于则减少左边一列
*注意：如果数组中有重复数字，那从左下角开始查找，只能查找距离左下角元素最近的那个数字。
*/
class Problem4_Solution1 {
    public boolean solve(int[][] arr, int num) {
        int row = arr.length; //获得二维数组的行数
        int col = arr[0].length; //获得二维数组的列数
        int find_row = row - 1;
        int find_col = 0;
        while((find_row >= 0) && (find_col <= col-1)) {
            if(arr[find_row][find_col] == num) {
                System.out.println("find it successfully...");
                System.out.println("查找数字是数组的第 " + (find_row+1) + " 行，第 "+ (find_col+1) +" 列");
                return true;
            } else if(arr[find_row][find_col] > num){
                find_row -= 1;
            } else {
                find_col += 1;
            }
        }

        return false;
    }
}

下面这个是结果图。从问题图中那个4x4的二维矩阵中查找出数字7。