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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章