面试题4：二维数组中的查找

题目

该系列文章题目和思路均参考自：《剑指Offer》

例如从下面的二维数组中查找数字7，则返回true，查找数字5，则返回false。

解法

具体思路：

• 首先选取数组中右上角的元素，如果该元素等于要查找的元素，则查找过程结束。
• 如果该数字大于要查找的数字，则剔除这个数字所在的列。
• 如果该数字小于要查找的数字，则剔除该数字所在的行。
• 如此进行，直到查找到这个数字或者查找范围为空。

代码实现

public class ON_FindInPartiallySortedMatrix {

    public static boolean findInPartiallySortedMatrix(int[][] array, int m, int n, int num) {
        if (array == null || m <= 0 || n <= 0)
            return false;
        int i = 0;
        int j = n - 1;
        while (i < m && j >= 0) {
            if (array[i][j] == num) {
                return true;
            } else if (array[i][j] > num) {
                j -= 1;
            } else {
                i += 1;
            }
        }
        return false;
    }

    public static void main(String[] args) {
        int[][] array = {{1, 2, 8, 9},
                {2, 4, 9, 12},
                {4, 7, 10, 13},
                {6, 8, 11, 15}};
        Utils.printMatrix(array, 4, 4);

        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 7));
        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 2));
        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 5));
        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 15));
        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 1));
        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 9));
        System.out.println(findInPartiallySortedMatrix(array, 4,  4, 6));
    }
}