剑指offer 面试题3 二维数组中的查找

剑指offer 面试题3 二维数组中的查找

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

/** * Created by liyazhou on 2017/5/22. */public class Test3 {    public static boolean contains(int[][] matrix, int number){        for (int i = 0; i < matrix.length; i++)            for (int j = 0; j < matrix[i].length; j++)                if (matrix[i][j] == number) {                    System.out.println(String.format("i, j = %d, %d", i, j));                    return true;                }        return false;    }    public static boolean contains2(int[][] matrix, int number){        boolean contains = false;        int col = matrix[0].length-1;        int row = 0; // 右上角元素的        if (number < matrix[0][0] || number > matrix[matrix[0].length-1][matrix.length-1]) return false;  // 判断边界        while(row < matrix.length && col >= 0){            int topRightVal = matrix[row][col];            if (number == topRightVal) {                contains = true; break;            }            else if (number < topRightVal) {  // 删除当前元素所在列                --col;            }else{ // number > topRightVal，删除当前元素所在行                ++row;            }        }        System.out.println(String.format("row, col = %d, %d", row, col));        return contains;    }    /**     * 在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。     * 请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。     * <p/>     * 规律：     *  首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束。     *  如果该数字大于要查找的数字，剔除这个数字所在的列；     *  如果该数字小于要查找的数字，剔除这个数字所在的行。     *  也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除行或者列，     *  这样每一步都可以缩小查找范围，直到要查找的数字，或者查找范围为空。     * @param matrix 二维数组     * @param number 待查找的数字     * @return 在数组中是否找到该数字     */    public static boolean contains3(int[][] matrix, int number){        if (matrix == null || matrix.length < 1 || matrix[0].length < 1) return false;  // 判断数组        int rows = matrix.length;        int cols = matrix[0].length;        if (number < matrix[0][0] || number > matrix[rows-1][cols-1]) return false;  // 判断边界        boolean found = false;        int row = 0;        int col = cols-1;  // todo, 注意边界        while(row < rows && col >=0){            int topRightVal = matrix[row][col];  // 获取到查找范围内的右上角的元素            if (number == topRightVal){                found = true; break;  // 找到该数字，停止查询            }            else if (number < topRightVal) --col;  // 删除当前列， 第 col 列            else ++row;  // number > topRightVal， 删除当前行，第 row 行        }        System.out.println(String.format("row, col = %d, %d", row, col));        return found;    }    public static void main(String[] args){        int[][] matrix = {                {1 ,2, 8, 9},                {2, 4, 9, 12},                {4, 7, 10, 13},                {6, 8, 11, 15}        };        int[] numbers = {-1, 1, 15, 7};        for (int number : numbers){            System.out.println(String.format("contains  %d, %b", number, Test3.contains(matrix, number)));            System.out.println(String.format("contains2 %d, %b", number, Test3.contains2(matrix, number)));            System.out.println(String.format("contains3 %d, %b", number, Test3.contains3(matrix, number)));        }    }}