二维数组中的查找

题目

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

解法

从右上角看整个二维数组是一个二叉排序树，从右上角开始，依次比较大小，

• 如果大于指定数则向左移一位
• 如果小于指定数则向下移一位
• 直至相等或者超过数组界限退出

代码实现

class Solution {
    public boolean findNumberIn2DArray(int[][] matrix, int target) {
        if (matrix == null || matrix.length == 0 || matrix[0].length == 0)
            return false;
        int c = 0;
        int r = matrix[0].length - 1;
        boolean res = true;
        while (matrix[c][r] != target) {
            if (matrix[c][r] < target)
                c += 1;
            else if (matrix[c][r] > target)
                r -= 1;
            if (c >= matrix.length || r< 0) {
                res = false;
                break;
            }
        }
        return res;
    }

    public static void main(String[] args) {
        int[][] m = {{1,   4,  7, 11, 15},
                    {2,   5,  8, 12, 19},
                    {3,   6,  9, 16, 22},
                    {10, 13, 14, 17, 24},
                    {18, 21, 23, 26, 30}};
        int t = 20;
        System.out.println(new Solution().findNumberIn2DArray(m,t));
    }
}