劍指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;
    }
}