剑指offer--二维数组的查找

package test;

import java.util.HashSet;
import java.util.Scanner;

public class SetTest {

    // 二维数组的查找

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

    // 思路： 依次比较右上角的数字；如果该数字大于要查找的数字，则剔除列；如果该数字大于要查找的数字，则剔除行；
    // 复杂度：O(m+n), 行数m，列数n
    public static void main(String[] args) {
        int[][] arrays = new int[3][3];
        arrays[0][0] = 1;
        arrays[0][1] = 2;
        arrays[0][2] = 3;
        arrays[1][0] = 4;
        arrays[1][1] = 5;
        arrays[1][2] = 6;
        arrays[2][0] = 7;
        arrays[2][1] = 8;
        arrays[2][2] = 9;
        Scanner scan = new Scanner(System.in);
        int num = scan.nextInt();
        boolean flag = find(arrays, num);
        if (flag) {
            System.out.println("存在这个数字");
        } else {
            System.out.println("不存在这个数字");
        }

    }

    public static boolean find(int[][] arrays, int num) {

        int rows = arrays.length;
        int cols = arrays[0].length;
        int i = rows - 1, j = 0;
        while (i >= 0 && j < cols) {
            if (num < arrays[i][j])
                i--;
            else if (num > arrays[i][j])
                j++;
            else
                return true;
        }
        return false;
    }
}